setup-python/__tests__/verify-python.py

# The sys module provides information about constants, functions and methods of the Python interpreter.
import sys
#The len() function returns the number of items in an object. When the object is a string, the len() function returns the number of characters in the string.
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")