def test_loading(self): # test root level directory logs = MachineLogs(self.logs_dir, recursive=False, verbose=True) self.assertEqual(logs.num_logs, 13) # test recursive logs = MachineLogs(self.logs_dir, verbose=False) self.assertEqual(logs.num_logs, 17) # test using method logs = MachineLogs() logs.load_dir(self.logs_dir, verbose=False) self.assertEqual(logs.num_logs, 17)