From 055e80cfadccc65d1eeb089f39fb68e3d5e1a9da Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Fri, 10 Jan 2020 21:36:18 +0000 Subject: [PATCH] rm old ConfigTester --- tests/test_modeling_common.py | 32 -------------------------------- 1 file changed, 32 deletions(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 719debcb3c..f5f93c7f07 100644 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -592,38 +592,6 @@ class ModelTesterMixin: model(**inputs_dict) -class ConfigTester(object): - def __init__(self, parent, config_class=None, **kwargs): - self.parent = parent - self.config_class = config_class - self.inputs_dict = kwargs - - def create_and_test_config_common_properties(self): - config = self.config_class(**self.inputs_dict) - self.parent.assertTrue(hasattr(config, "vocab_size")) - self.parent.assertTrue(hasattr(config, "hidden_size")) - self.parent.assertTrue(hasattr(config, "num_attention_heads")) - self.parent.assertTrue(hasattr(config, "num_hidden_layers")) - - def create_and_test_config_to_json_string(self): - config = self.config_class(**self.inputs_dict) - obj = json.loads(config.to_json_string()) - for key, value in self.inputs_dict.items(): - self.parent.assertEqual(obj[key], value) - - def create_and_test_config_to_json_file(self): - config_first = self.config_class(**self.inputs_dict) - json_file_path = os.path.join(os.getcwd(), "config_" + str(uuid.uuid4()) + ".json") - config_first.to_json_file(json_file_path) - config_second = self.config_class.from_json_file(json_file_path) - os.remove(json_file_path) - self.parent.assertEqual(config_second.to_dict(), config_first.to_dict()) - - def run_common_tests(self): - self.create_and_test_config_common_properties() - self.create_and_test_config_to_json_string() - self.create_and_test_config_to_json_file() - global_rng = random.Random()