Add examples with unittest.TestCase.

ionelmc · Feb 6, 2015 · d8035c3 · d8035c3
1 parent 6656e7f
commit d8035c3
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/test_with_testcase.py b/tests/test_with_testcase.py
@@ -0,0 +1,23 @@
+import time
+import unittest
+
+import pytest
+
+
+class TerribleTerribleWayToWriteTests(unittest.TestCase):
+    @pytest.fixture(autouse=True)
+    def setupBenchmark(self, benchmark):
+        self.benchmark = benchmark
+
+    def test_foo(self):
+        self.benchmark(time.sleep, 0.000001)
+
+
+class TerribleTerribleWayToWritePatchTests(unittest.TestCase):
+    @pytest.fixture(autouse=True)
+    def setupBenchmark(self, benchmark_weave):
+        self.benchmark_weave = benchmark_weave
+
+    def test_foo2(self):
+        with self.benchmark_weave('time.sleep'):
+            time.sleep(0.0000001)