def test_close_file(self): logger = Logger(xlabel="x", ylabel="y", legend="test", log_path="./newtest/test_log.txt", csv_path="./newtest/test_csv.csv") logger.close_file() self.assertTrue(os.path.exists('./newtest/'))
def test_log(self): logger = Logger(xlabel="x", ylabel="y", legend="test", log_path="./newtest/test_log.txt") logger.log("test text") f = open("./newtest/test_log.txt", "r") contents = f.read() self.assertEqual(contents, "test text\n") logger.close_file()
def test_log(self): log_path = "./newtest/test_log.txt" log_dir = os.path.dirname(log_path) if os.path.exists(log_dir): shutil.rmtree(log_dir) logger = Logger(xlabel="x", ylabel="y", legend="test", log_path=log_path) logger.log("test text") f = open("./newtest/test_log.txt", "r") contents = f.read() self.assertEqual(contents, "test text\n") logger.close_file() shutil.rmtree(log_dir)