def test_difference_update_nonexistent(self): versions = Versions() versions.add('1.0') versions.add('2.0') versions.add('3.0', aliases=['latest']) with self.assertRaises(KeyError): versions.difference_update(['1.0', 'latest', '4.0']) self.assertEqual(list(versions), [ VersionInfo('3.0', aliases=['latest']), VersionInfo('2.0'), VersionInfo('1.0'), ])
def test_difference_update(self): versions = Versions() versions.add('1.0') versions.add('2.0') versions.add('3.0', aliases=['latest']) v = versions.difference_update(['1.0', '2.0', 'latest']) self.assertEqual(v, [VersionInfo('1.0'), VersionInfo('2.0'), 'latest']) self.assertEqual(list(versions), [ VersionInfo('3.0'), ])