def test_loading(self): # test root level directory logs = MachineLogs(self.logs_dir, recursive=False) self.assertEqual(logs.num_logs, 3) # test recursive logs = MachineLogs(self.logs_dir) self.assertEqual(logs.num_logs, 3) # test using zip file zfile = osp.join(self._logs_dir, 'mixed_types.zip') logs = MachineLogs.from_zip(zfile) self.assertEqual(logs.num_logs, 3)
def test_loading(self): # test root level directory logs = MachineLogs(self.logs_dir, recursive=False) self.assertEqual(logs.num_logs, 3) # test recursive logs = MachineLogs(self.logs_dir) self.assertEqual(logs.num_logs, 3) # test using zip file zfile = get_file_from_cloud_test_repo(['mlc_logs', 'mixed_types.zip']) logs = MachineLogs.from_zip(zfile) self.assertEqual(logs.num_logs, 3)