def test_collect_tensorflow_environment_variables(self): os.environ["TF_ENABLE_WINOGRAD_NONFUSED"] = "1" os.environ["TF_OTHER"] = "2" os.environ["OTHER"] = "3" run_info = {} logger._collect_tensorflow_environment_variables(run_info) self.assertIsNotNone(run_info["tensorflow_environment_variables"]) expected_tf_envs = [ {"name": "TF_ENABLE_WINOGRAD_NONFUSED", "value": "1"}, {"name": "TF_OTHER", "value": "2"}, ] self.assertEqual(run_info["tensorflow_environment_variables"], expected_tf_envs)
def test_collect_tensorflow_environment_variables(self): os.environ["TF_ENABLE_WINOGRAD_NONFUSED"] = "1" os.environ["TF_OTHER"] = "2" os.environ["OTHER"] = "3" run_info = {} logger._collect_tensorflow_environment_variables(run_info) self.assertIsNotNone(run_info["tensorflow_environment_variables"]) expected_tf_envs = [ {"name": "TF_ENABLE_WINOGRAD_NONFUSED", "value": "1"}, {"name": "TF_OTHER", "value": "2"}, ] self.assertEqual(run_info["tensorflow_environment_variables"], expected_tf_envs)