예제 #1
0
 def test_remove_abbr(self):
     # Set up test objects
     test_repo = ConvertRepo()
     test_type = ConvertType(test_repo, "test_type")
     test_type.base_unit = ConvertUnit(test_type, ["base_unit"], 1)
     test_unit_names = ["name1", "name2"]
     test_value = 1337
     test_unit = ConvertUnit(test_type, test_unit_names, test_value)
     test_unit.add_abbr("abbr1")
     test_unit.add_abbr("abbr2")
     # Test setup
     assert "abbr1" in test_unit.abbr_list
     assert "abbr2" in test_unit.abbr_list
     assert len(test_unit.abbr_list) == 2
     # Remove abbreviation
     test_unit.remove_abbr("abbr1")
     # Test changes
     assert len(test_unit.abbr_list) == 1
     assert "abbr2" in test_unit.abbr_list
     assert "abbr1" not in test_unit.abbr_list