def test_list_of_tuples(self): version_map = [ (LooseVersion('1'), 'foo'), (LooseVersion('2'), 'bar'), (LooseVersion('3'), 'baz'), ] self.assertEqual(_map_version(version_map, '1.1'), 'foo') self.assertEqual(_map_version(version_map, '2'), 'bar') self.assertEqual(_map_version(version_map, '4.5'), 'baz') self.assertEqual(_map_version(version_map, '11.11'), 'baz') self.assertEqual(_map_version(version_map, '0.1'), 'foo')
def test_dict(self): version_map = { '1': 'foo', '2': 'bar', '3': 'baz', } self.assertEqual(_map_version(version_map, '1.1'), 'foo') # test exact match self.assertEqual(_map_version(version_map, '2'), 'bar') # versions are just minimums self.assertEqual(_map_version(version_map, '4.5'), 'baz') # compare versions, not strings self.assertEqual(_map_version(version_map, '11.11'), 'baz') # fall back to lowest version self.assertEqual(_map_version(version_map, '0.1'), 'foo')