def testPathSeparator(self): # In nix systems, path separator would be the same as that of CNS/GCS # making it hard to tell if something went wrong. self.stubs.Set(os, "sep", "#") self.assertEqual(io_wrapper.PathSeparator("/tmp/foo"), "#") self.assertEqual(io_wrapper.PathSeparator("tmp/foo"), "#") self.assertEqual(io_wrapper.PathSeparator("/cns/tmp/foo"), "/") self.assertEqual(io_wrapper.PathSeparator("gs://foo"), "/")
def testPathSeparator(self): # In nix systems, path separator would be the same as that of CNS/GCS # making it hard to tell if something went wrong. self.stubs.Set(os, 'sep', '#') self.assertEqual(io_wrapper.PathSeparator('/tmp/foo'), '#') self.assertEqual(io_wrapper.PathSeparator('tmp/foo'), '#') self.assertEqual(io_wrapper.PathSeparator('/cns/tmp/foo'), '/') self.assertEqual(io_wrapper.PathSeparator('gs://foo'), '/')
def _get_exp_and_run_names(self, path, subdir, experiment_name_override=None): if experiment_name_override is not None: return (experiment_name_override, os.path.relpath(subdir, path)) sep = io_wrapper.PathSeparator(path) path_parts = os.path.relpath(subdir, path).split(sep, 1) experiment_name = path_parts[0] run_name = path_parts[1] if len(path_parts) == 2 else '.' return (experiment_name, run_name)