use skipif

oturns · Oct 2, 2023 · d917836 · d917836
1 parent be47941
commit d917836
Showing 1 changed file with 6 additions and 17 deletions.
diff --git a/geosnap/tests/test_constructors.py b/geosnap/tests/test_constructors.py
@@ -24,26 +24,15 @@ def test_nces_school_dists():
     dists = io.get_nces(store, dataset="school_districts")
     assert dists.shape == (13352, 18)
 
-
+@pytest.mark.skipif(sys.platform.startswith("win"), reason="skipping test on windows due to mem failure")
 def test_ejscreen():
-    if sys.platform.startswith("win"):
-        pytest.skip(
-            "skipping test on windows due to mem failure", allow_module_level=True
-        )
-
-    else:
-        ej = io.get_ejscreen(store, years=[2018], fips=["11"])
-        assert ej.shape == (450, 369)
-
+    ej = io.get_ejscreen(store, years=[2018], fips=["11"])
+    assert ej.shape == (450, 369)
 
+@pytest.mark.skipif(sys.platform.startswith("win"), reason="skipping test on windows due to mem failure")
 def test_nces_sabs():
-    if sys.platform.startswith("win"):
-        pytest.skip(
-            "skipping test on windows due to mem failure", allow_module_level=True
-        )
-    else:
-        sabs = io.get_nces(store, dataset="sabs")
-        assert sabs.shape == (75128, 15)
+    sabs = io.get_nces(store, dataset="sabs")
+    assert sabs.shape == (75128, 15)
 
 
 def test_acs():