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