def testReadEnvYaml(self): env = {'hello': 'world', 'foo': 'bar'} output_dir = self.CreateTempDir() util.WriteEnvYaml(env, output_dir) yaml_env = util.ReadEnvYaml(output_dir) self.assertEqual(env, yaml_env)
def Run(self, args): return util.ReadEnvYaml(args.data_dir)
def Run(self, args): data_dir = bigtable_util.GetDataDir() return util.ReadEnvYaml(data_dir)
def Run(self, args): data_dir = spanner_util.GetDataDir() return util.ReadEnvYaml(data_dir)
def testReadNonExistentEnvYaml(self): output_dir = self.CreateTempDir() with self.assertRaises(util.NoEnvYamlError): util.ReadEnvYaml(output_dir)