class TestStringSet(unittest.TestCase): """Test stringset class.""" def setUp(self): self.stringset = StringSet() def test_correct_addition(self): """Test that entries are correctly added to a StringSet.""" t1 = GenericTranslation('se1', 'trans1', context='') t2 = GenericTranslation('se2', 'trans2', context='') t3 = GenericTranslation('se2', 'trans1', context='c3') translations = [t1, t2, t3, ] for idx, t in enumerate(translations, start=1): self.stringset.add(t) self.assertEquals(len(self.stringset), idx) for old_t in translations[:idx]: self.assertIn(old_t, self.stringset) def test_adding_duplicates(self): """Test that adding a duplicate element only keeps one.""" t1 = GenericTranslation('se', 'trans1', context='') self.stringset.add(t1) self.assertEquals(len(self.stringset), 1) self.assertIn(t1, self.stringset) t2 = GenericTranslation('se', 'trans2', context='') self.stringset.add(t2) self.assertEquals(len(self.stringset), 1) self.assertIn(t2, self.stringset) for t in self.stringset: self.assertEqual(t1.translation, t.translation)
def setUp(self): self.stringset = StringSet()