Merge pull request #8 from eldaduzman/fix-decorator-sig

fix decorator signature

Author: eldaduzman

.gitignore

@@ -337,3 +337,4 @@ $RECYCLE.BIN/
 # Custom rules (everything added below won't be overriden by 'Generate .gitignore File' if you use 'Update' option)
 
 .vscode/
+localstuff/

README.md

@@ -29,7 +29,7 @@ Decorate your unittest class with `@perf_unit_test_class` to turn standard unit
 from perf_unit import perf_unit_test_class
 import unittest
 
-@perf_unit_test_class()
+@perf_unit_test_class
 class MyTestCase(unittest.TestCase):
 
     def test_example1(self):

src/perf_unit/__init__.py

@@ -61,6 +61,7 @@ def run_single_iteration(
 
 
 def perf_unit_test_class(
+    *args,
     how_many_threads: int = 30,
     total_number_of_method_executions: int = 100,
     upper_median_threashold_in_milliseconds: int = 500,
@@ -91,7 +92,7 @@ def wrapped_method(*args, **kwargs):
                             )
                         )
 
-                execution_times = [future.result() for future in futures]
+                execution_times = tuple([future.result() for future in futures])
 
                 median_time = statistics.median(execution_times)
 
@@ -121,4 +122,7 @@ def wrapped_method(*args, **kwargs):
 
         return cls
 
+    if len(args) == 1:
+        return modify_test_class(args[0])
+
     return modify_test_class

utests/test_basic.py

@@ -21,7 +21,9 @@ def test_method(false_self):
         self.assertTrue(result.wasSuccessful())
 
     def test_should_fail_due_too_high_latency(self):
-        @perf_unit_test_class(how_many_threads=50, upper_median_threashold_in_milliseconds=1)
+        @perf_unit_test_class(
+            how_many_threads=50, upper_median_threashold_in_milliseconds=1
+        )
         class SomeTestClass(TestCase):
             def test_method(false_self):
                 time.sleep(random.randint(200, 600) / 1000)
@@ -50,6 +52,19 @@ def test_class_has_no_test_methods(self):
         class SomeTestClass(TestCase):
             pass
 
+    def test_decorators_with_no_inputs(self):
+        @perf_unit_test_class
+        class SomeTestClass(TestCase):
+            def test_method(false_self):
+                pass
+
+        suite = TestLoader().loadTestsFromTestCase(SomeTestClass)
+
+        runner = TextTestRunner()
+        result = runner.run(suite)
+        result.printErrors()
+        self.assertTrue(result.wasSuccessful())
+
 
 if __name__ == "__main__":
     main()