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)