runner-images/images/macos/scripts/tests/Python.Tests.ps1

Import-Module "$PSScriptRoot/../helpers/Common.Helpers.psm1"
Import-Module "$PSScriptRoot/Helpers.psm1" -DisableNameChecking

$os = Get-OSVersion

Describe "Python3" {
    It "Python 3 is available" {
        "python3 --version" | Should -ReturnZeroExitCode
    }
    
    if ($os.IsVenturaArm64 -or $os.IsSonomaArm64 -or $os.IsSequoiaArm64) {
        It "Python 3 is installed under /opt/homebrew/bin/" {
            Get-ToolPath "python3" | Should -BeLike "/opt/homebrew/bin/*"
        }
    } else {
        It "Python 3 is installed under /usr/local/bin" {
            Get-ToolPath "python3" | Should -BeLike "/usr/local/bin*"
        }
    }

    It "Pip 3 is available" {
        "pip3 --version" | Should -ReturnZeroExitCode
    }

    It "Pipx is available" {
        "pipx --version" | Should -ReturnZeroExitCode
    }

    It "Pip 3 and Python 3 came from the same brew formula" {
        $pip3Path = Split-Path (readlink (which pip3))
        $python3Path = Split-Path (readlink (which python3))
        $pip3Path | Should -BeExactly $python3Path
    }

}

Describe "Python2" -Skip:($os.IsVentura -or $os.IsSonoma -or $os.IsSequoia) {
    It "Python 2 is available" {
        "/Library/Frameworks/Python.framework/Versions/2.7/bin/python --version" | Should -ReturnZeroExitCode
    }

    It "Pip 2 is available" {
        "/Library/Frameworks/Python.framework/Versions/2.7/bin/pip --version" | Should -ReturnZeroExitCode
    }

    It "2to3 symlink does not point to Python 2" {
        $2to3path = (Get-ChildItem (Get-Command 2to3).Path).Target
        $2to3path | Should -Not -BeLike '/Frameworks/Python.framework/Versions/2.*'
    }
}