Exemple #1
0
    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'))
Exemple #2
0
    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)