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)
