From 28cc8fe90c72ec6937883dccd6f72fe018ab139c Mon Sep 17 00:00:00 2001
From: Prodesire <prodesire@yeah.net>
Date: Sat, 21 Apr 2018 20:55:10 +0800
Subject: [PATCH] add test for dt.timer

---
 tests/test_dt.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 tests/test_dt.py

diff --git a/tests/test_dt.py b/tests/test_dt.py
new file mode 100644
index 0000000..474bd8d
--- /dev/null
+++ b/tests/test_dt.py
@@ -0,0 +1,23 @@
+import os
+from pydu.dt import timer
+
+
+class TestTimer(object):
+    def test_context_manager(self):
+        timeit = timer()
+
+        with timeit:
+            os.getcwd()
+
+        assert timeit.elapsed is not None
+
+    def test_decorator(self):
+        import sys
+        timeit = timer(print_func=print)
+
+        @timeit
+        def foo():
+            os.getcwd()
+
+        foo()
+        assert timeit.elapsed is not None