Fix linting and tests neglected from last commit

jessebrennan · May 11, 2020 · 30a5bc3 · 30a5bc3
1 parent 41cb601
commit 30a5bc3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/src/cutest/__init__.py b/src/cutest/__init__.py
@@ -63,11 +63,7 @@ def fixture(self, obj) -> 'FixtureDefinition':
         block. Any cleanup should happen inside of an attached finally
         block.
         """
-        if (
-                inspect.isclass(obj)
-                and hasattr(obj, '__enter__')
-                and hasattr(obj, '__exit__')
-        ):
+        if inspect.isclass(obj) and hasattr(obj, '__enter__') and hasattr(obj, '__exit__'):
             return FixtureDefinition(self, obj)
         elif inspect.isgeneratorfunction(obj):
             cm = contextmanager(obj)

diff --git a/test/test_runner.py b/test/test_runner.py
@@ -7,7 +7,8 @@
 class RunnerTest(unittest.TestCase):
 
     def test_prune(self):
-        runner = cutest.SerialRunner()
+        test_output = cutest.default_output_stream()
+        runner = cutest.SerialRunner(test_output, verbosity=1)
         sample.my_suite.initialize()
         pruned = list(runner.pruned_suites(sample.test_1.calls))
         self.assertEqual(len(pruned), 1)