def test_get_translations(self): # List of Equivalent instances is empty self.assertListEqual(self.sense.get_translations(), []) # Create Equivalent instances and add them to the list equivalent1 = Equivalent() equivalent2 = Equivalent() self.sense.equivalent = [equivalent1, equivalent2] # Set their translations trans1 = "trans1" trans2 = "trans2" equivalent1.translation = trans1 # Test get translations self.assertListEqual(self.sense.get_translations(), [trans1]) equivalent2.translation = trans2 self.assertListEqual(self.sense.get_translations(), [trans1, trans2]) # Test with a language filter lang = "lang" equivalent2.language = lang self.assertListEqual(self.sense.get_translations(), [trans1, trans2]) self.assertListEqual(self.sense.get_translations("eng"), []) self.assertListEqual(self.sense.get_translations(lang), [trans2]) # Delete Equivalent instances del self.sense.equivalent[:] del equivalent1, equivalent2