From f030a903ba079acee1be93de59bab75ca90e329a Mon Sep 17 00:00:00 2001 From: superjom Date: Wed, 29 Nov 2017 19:18:32 +0800 Subject: [PATCH] add write and read service test --- visualdl/backend/python/test_read_service.py | 15 +++++++++++++++ visualdl/backend/python/test_write_service.py | 17 +++++++++++++++++ 2 files changed, 32 insertions(+) create mode 100644 visualdl/backend/python/test_read_service.py create mode 100644 visualdl/backend/python/test_write_service.py diff --git a/visualdl/backend/python/test_read_service.py b/visualdl/backend/python/test_read_service.py new file mode 100644 index 00000000..0831e04b --- /dev/null +++ b/visualdl/backend/python/test_read_service.py @@ -0,0 +1,15 @@ +import summary +import numpy as np +import unittest +import time + + +class StorageTester(unittest.TestCase): + def test_storage(self): + summary.set_writable_storage("./tmp_dir") + time.sleep(5) + summary.stop_service() + + +if __name__ == '__main__': + unittest.main() diff --git a/visualdl/backend/python/test_write_service.py b/visualdl/backend/python/test_write_service.py new file mode 100644 index 00000000..241869c4 --- /dev/null +++ b/visualdl/backend/python/test_write_service.py @@ -0,0 +1,17 @@ +import summary +import numpy as np +import unittest +import time + + +class StorageTester(unittest.TestCase): + def test_read_storage(self): + summary.set_readable_storage("./tmp") + time.sleep(1) + scalar = summary.read_scalar('tag01') + time.sleep(5) + summary.stop_service() + + +if __name__ == '__main__': + unittest.main() -- GitLab