def test_flags_undefined(self): tpu_test_wrapper.maybe_define_flags() self.assertIn('tpu', flags.FLAGS) self.assertIn('zone', flags.FLAGS) self.assertIn('project', flags.FLAGS) self.assertIn('model_dir', flags.FLAGS)
def test_set_random_test_dir(self): tpu_test_wrapper.maybe_define_flags() tpu_test_wrapper.set_random_test_dir() self.assertStartsWith(flags.FLAGS.model_dir, 'gs://example-bucket/tempfiles') self.assertGreater(len(flags.FLAGS.model_dir), len('gs://example-bucket/tempfiles'))
def test_set_random_test_dir_repeatable(self): tpu_test_wrapper.maybe_define_flags() tpu_test_wrapper.set_random_test_dir() first = flags.FLAGS.model_dir tpu_test_wrapper.set_random_test_dir() second = flags.FLAGS.model_dir self.assertNotEqual(first, second)
def test_flags_already_defined_not_overridden(self): flags.DEFINE_string('tpu', 'tpuname', 'helpstring') tpu_test_wrapper.maybe_define_flags() self.assertIn('tpu', flags.FLAGS) self.assertIn('zone', flags.FLAGS) self.assertIn('project', flags.FLAGS) self.assertIn('model_dir', flags.FLAGS) self.assertEqual(flags.FLAGS.tpu, 'tpuname')