def test_remove_name(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 setup assert test_unit.name_list == test_unit_names assert len(test_unit.name_list) == 2 # Remove name test_unit.remove_name("name1") # Test changes assert len(test_unit.name_list) == 1 assert "name2" in test_unit.name_list assert "name1" not in test_unit.name_list