Fix marshmallow test helpers so that they work will all unittest comp…

…atible frameworks and not just pytest. 'python setup.py test' works again. (#127)
plangrid · Sep 17, 2019 · 8b30e5a · 8b30e5a
1 parent f836681
commit 8b30e5a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/helpers.py b/tests/helpers.py
@@ -1,11 +1,10 @@
-import pytest
-
+import unittest
 from flask_rebar.compat import MARSHMALLOW_V2, MARSHMALLOW_V3
 
 
-skip_if_marshmallow_not_v2 = pytest.mark.skipif(
+skip_if_marshmallow_not_v2 = unittest.skipIf(
     not MARSHMALLOW_V2, reason="Only applicable for Marshmallow version 2"
 )
-skip_if_marshmallow_not_v3 = pytest.mark.skipif(
+skip_if_marshmallow_not_v3 = unittest.skipIf(
     not MARSHMALLOW_V3, reason="Only applicable for Marshmallow version 3"
 )