def test_get_versions_from_classifiers():
assert get_versions_from_classifiers([
'Development Status :: 4 - Beta',
'Environment :: Console',
'Intended Audience :: Developers',
'License :: OSI Approved :: GNU General Public License (GPL)',
'Operating System :: OS Independent',
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: Implementation :: CPython',
'Programming Language :: Python :: Implementation :: PyPy',
]) == ['2.7', '3.6', '3.7', 'PyPy']
def test_get_versions_from_classifiers_with_only_suffix():
assert get_versions_from_classifiers([
'Programming Language :: Python :: 2 :: Only',
]) == ['2']
def test_get_versions_from_classifiers_with_trailing_whitespace():
# I was surprised too that this is allowed!
assert get_versions_from_classifiers([
'Programming Language :: Python :: 3.6 ',
]) == ['3.6']
def test_get_versions_from_classifiers_major_only():
assert get_versions_from_classifiers([
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 3',
]) == ['2', '3']