def test_parse_pb_file(self):
"""Test parse pb file success."""
filename = 'ms_output.pb'
create_graph_pb_file(output_dir=self._summary_dir, filename=filename)
parser = _PbParser(self._summary_dir)
tensor_event = parser._parse_pb_file(filename)
assert isinstance(tensor_event, TensorEvent)
def test_sort_pb_file_by_mtime(self):
"""Test sort pb files."""
filenames = ['abc.pb', 'bbc.pb']
for file in filenames:
create_graph_pb_file(output_dir=self._summary_dir, filename=file)
parser = _PbParser(self._summary_dir)
sorted_filenames = parser.sort_files(filenames)
assert filenames == sorted_filenames
def test_set_latest_file(self):
"""Test set latest file."""
filename = 'ms_output.pb'
create_graph_pb_file(output_dir=self._summary_dir, filename=filename)
parser = _PbParser(self._summary_dir)
is_latest = parser._set_latest_file(filename)
assert is_latest
filename = 'ms_output_older.pb'
file_path = create_graph_pb_file(output_dir=self._summary_dir,
filename=filename)
atime = 1
mtime = 1
os.utime(file_path, (atime, mtime))
is_latest = parser._set_latest_file(filename)
assert not is_latest
def test_sort_pb_file_by_filename(self):
"""Test sort pb file by file name."""
filenames = ['aaa.pb', 'bbb.pb', 'ccc.pb']
for file in filenames:
create_graph_pb_file(output_dir=self._summary_dir, filename=file)
atime, mtime = (3, 3)
os.utime(os.path.realpath(os.path.join(self._summary_dir, 'aaa.pb')),
(atime, mtime))
atime, mtime = (1, 1)
os.utime(os.path.realpath(os.path.join(self._summary_dir, 'bbb.pb')),
(atime, mtime))
os.utime(os.path.realpath(os.path.join(self._summary_dir, 'ccc.pb')),
(atime, mtime))
expected_filenames = ['bbb.pb', 'ccc.pb', 'aaa.pb']
parser = _PbParser(self._summary_dir)
sorted_filenames = parser.sort_files(filenames)
assert expected_filenames == sorted_filenames
