Use unittest pytestmark for factory method tests

vyahello · Oct 14, 2020 · 3917a25 · 3917a25
1 parent 5e050b5
commit 3917a25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/creational/test_factory_method.py b/tests/creational/test_factory_method.py
@@ -13,6 +13,8 @@
 )
 from tests.marker import unittest
 
+pytestmark = unittest
+
 
 @pytest.fixture(scope="module")
 def circle() -> Shape:
@@ -24,12 +26,10 @@ def square() -> Shape:
     return Square()
 
 
-@unittest
 def test_draw_circle(circle: Shape) -> None:
     assert circle.draw() == "Circle.draw"
 
 
-@unittest
 def test_draw_square(square: Shape) -> None:
     assert square.draw() == "Square.draw"