def test_change_custom_name(self): replace_custom_layer_name(self.spec, 'name1', 'notname1') names = get_custom_layer_names(self.spec) self.assertEqual(names, {'notname1', 'name2'}) #set it back for future tests replace_custom_layer_name(self.spec, 'notname1', 'name1')
def test_get_custom_names(self): names = get_custom_layer_names(self.spec) self.assertEqual(names, {'name1', 'name2'})