3
0
Fork 0
mirror of https://gitea.com/actions/setup-python.git synced 2024-11-25 19:49:35 +01:00
setup-python/__tests__/verify-python.py
Konrad Pabjan 6cbb9cfdfd
Improve workflows (#88)
* Improve workflows

* Update workflows

* Small fix
2020-05-01 18:52:58 +02:00

22 lines
No EOL
1,000 B
Python

import sys
argCount = len(sys.argv) - 1
if argCount == 1:
expectedVersion = sys.argv[1]
versions = len(expectedVersion.split("."))
majorMinor = str(sys.version_info[0]) + '.' + str(sys.version_info[1])
if versions == 2:
# Test only major and minor version
if expectedVersion != majorMinor:
raise Exception("Incorrect major + minor version detected\nExpected: " + expectedVersion + "\nActual: " + majorMinor)
elif versions == 3:
# Test major, minor and micro version
majorMinorMicro = majorMinor + '.' + str(sys.version_info[2])
if expectedVersion != majorMinorMicro:
raise Exception("Incorrect major + minor + micro version detected\nExpected: " + expectedVersion + "\nActual: " + majorMinorMicro)
else:
raise Exception("Incorrect number of arguments supplied")
print("Correct version of Python " + expectedVersion + " detected")
else:
raise Exception("Incorrect number of arguments supplied")