parameterized pytest functions to support local server

Author: jpswinski

tests/conftest.py

@@ -0,0 +1,26 @@
+import pytest
+
+def pytest_addoption(parser):
+    parser.addoption("--server", action="store", default="icesat2sliderule.org")
+    parser.addoption("--asset", action="store", default="nsidc-s3")
+
+#def pytest_generate_tests(metafunc):
+#    # This is called for every test. Only get/set command line arguments
+#    # if the argument is specified in the list of test "fixturenames".
+#    option_value = metafunc.config.option.server
+#    if 'server' in metafunc.fixturenames and option_value is not None:
+#        metafunc.parametrize("server", [option_value])
+
+@pytest.fixture(scope='session')
+def server(request):
+    server_value = request.config.option.server
+    if server_value is None:
+        pytest.skip()
+    return server_value
+
+@pytest.fixture(scope='session')
+def asset(request):
+    asset_value = request.config.option.asset
+    if asset_value is None:
+        pytest.skip()
+    return asset_value

tests/test_api.py

@@ -1,13 +1,9 @@
 """Tests for sliderule-python icesat2 api."""
 
 import pytest
-from requests.exceptions import ConnectTimeout, ConnectionError
 import sliderule
 from sliderule import icesat2
-from pathlib import Path
-import os.path
 
-SERVER = "icesat2sliderule.org"
 ASSET = "nsidc-s3"
 
 ATL03_FILE1 = "ATL03_20181019065445_03150111_004_01.h5"
@@ -20,8 +16,8 @@
 
 @pytest.mark.network
 class TestApi:
-    def test_time(self):
-        icesat2.init(SERVER)
+    def test_time(self, server):
+        icesat2.init(server)
         rqst = {
             "time": "NOW",
             "input": "NOW",
@@ -39,37 +35,37 @@ def test_time(self):
         again = d["time"]
         assert now == again
 
-    def test_h5(self):
-        icesat2.init(SERVER)
-        epoch_offset = icesat2.h5("ancillary_data/atlas_sdp_gps_epoch", ATL03_FILE1, ASSET)[0]
+    def test_h5(self, server, asset):
+        icesat2.init(server)
+        epoch_offset = icesat2.h5("ancillary_data/atlas_sdp_gps_epoch", ATL03_FILE1, asset)[0]
         assert epoch_offset == 1198800018.0
 
-    def test_h5_types(self):
-        icesat2.init(SERVER)
-        heights_64 = icesat2.h5("/gt1l/land_ice_segments/h_li", ATL06_FILE1, ASSET)
+    def test_h5_types(self, server, asset):
+        icesat2.init(server)
+        heights_64 = icesat2.h5("/gt1l/land_ice_segments/h_li", ATL06_FILE1, asset)
         expected_64 = [45.95665, 45.999374, 46.017857, 46.015575, 46.067562, 46.099796, 46.14037, 46.105526, 46.096024, 46.12297]
-        heights_32 = icesat2.h5("/gt1l/land_ice_segments/h_li", ATL06_FILE2, ASSET)
+        heights_32 = icesat2.h5("/gt1l/land_ice_segments/h_li", ATL06_FILE2, asset)
         expected_32 = [350.46988, 352.08688, 352.43243, 353.19345, 353.69543, 352.25998, 350.15366, 346.37888, 342.47903, 341.51]
-        bckgrd_32nf = icesat2.h5("/gt1l/bckgrd_atlas/bckgrd_rate", ATL03_FILE2, ASSET)
+        bckgrd_32nf = icesat2.h5("/gt1l/bckgrd_atlas/bckgrd_rate", ATL03_FILE2, asset)
         expected_32nf = [29311.684, 6385.937, 6380.8413, 28678.951, 55349.168, 38201.082, 19083.434, 38045.67, 34942.434, 38096.266]
         for c in zip(heights_64, expected_64, heights_32, expected_32, bckgrd_32nf, expected_32nf):
             assert (round(c[0]) == round(c[1])) and (round(c[2]) == round(c[3])) and (round(c[4]) == round(c[5]))
 
-    def test_variable_length(self):
-        icesat2.init(SERVER)
-        v = icesat2.h5("/gt1r/geolocation/segment_ph_cnt", ATL03_FILE1, ASSET)
+    def test_variable_length(self, server, asset):
+        icesat2.init(server)
+        v = icesat2.h5("/gt1r/geolocation/segment_ph_cnt", ATL03_FILE1, asset)
         assert v[0] == 258 and v[1] == 256 and v[2] == 273
 
-    def test_h5p(self):
-        icesat2.init(SERVER)
+    def test_h5p(self, server, asset):
+        icesat2.init(server)
         datasets = [
             {"dataset": "/gt1l/land_ice_segments/h_li", "numrows": 5},
             {"dataset": "/gt1r/land_ice_segments/h_li", "numrows": 5},
             {"dataset": "/gt2l/land_ice_segments/h_li", "numrows": 5},
             {"dataset": "/gt2r/land_ice_segments/h_li", "numrows": 5},
             {"dataset": "/gt3l/land_ice_segments/h_li", "numrows": 5},
             {"dataset": "/gt3r/land_ice_segments/h_li", "numrows": 5} ]
-        rsps = icesat2.h5p(datasets, ATL06_FILE1, ASSET)
+        rsps = icesat2.h5p(datasets, ATL06_FILE1, asset)
         expected = {'/gt1l/land_ice_segments/h_li': [45.95665, 45.999374, 46.017857, 46.015575, 46.067562],
                     '/gt1r/land_ice_segments/h_li': [45.980865, 46.02602, 46.02262, 46.03137, 46.073578],
                     '/gt2l/land_ice_segments/h_li': [45.611526, 45.588196, 45.53242, 45.48105, 45.443752],
@@ -80,10 +76,10 @@ def test_h5p(self):
             for index in range(len(expected[dataset])):
                 assert round(rsps[dataset][index]) == round(expected[dataset][index])
 
-    def test_geospatial1(self):
-        icesat2.init(SERVER)
+    def test_geospatial1(self, server, asset):
+        icesat2.init(server)
         test = {
-            "asset": ASSET,
+            "asset": asset,
             "pole": "north",
             "lat": 40.0,
             "lon": 60.0,
@@ -125,10 +121,10 @@ def test_geospatial1(self):
         assert d["lat"] == 40.0 and d["lon"] == 60.0
         assert d["x"] == 0.466307658155 and d["y"] == 0.80766855588292
 
-    def test_geospatial2(self):
-        icesat2.init(SERVER)
+    def test_geospatial2(self, server, asset):
+        icesat2.init(server)
         test = {
-            "asset": ASSET,
+            "asset": asset,
             "pole": "north",
             "lat": 30.0,
             "lon": 100.0,
@@ -138,10 +134,10 @@ def test_geospatial2(self):
         d = sliderule.source("geo", test)
         assert abs(d["lat"] - 30.0) < 0.0001 and d["lon"] == 100.0
 
-    def test_geospatial3(self):
-        icesat2.init(SERVER)
+    def test_geospatial3(self, server, asset):
+        icesat2.init(server)
         test = {
-            "asset": ASSET,
+            "asset": asset,
             "pole": "north",
             "lat": 30.0,
             "lon": 100.0,
@@ -151,10 +147,10 @@ def test_geospatial3(self):
         d = sliderule.source("geo", test)
         assert abs(d["lat"] - 30.0) < 0.0001 and d["lon"] == -100.0
 
-    def test_geospatial4(self):
-        icesat2.init(SERVER)
+    def test_geospatial4(self, server, asset):
+        icesat2.init(server)
         test = {
-            "asset": ASSET,
+            "asset": asset,
             "pole": "north",
             "lat": 30.0,
             "lon": 100.0,
@@ -164,16 +160,16 @@ def test_geospatial4(self):
         d = sliderule.source("geo", test)
         assert abs(d["lat"] - 30.0) < 0.0001 and d["lon"] == -80.0
 
-    def test_definition(self):
-        icesat2.init(SERVER)
+    def test_definition(self, server):
+        icesat2.init(server)
         rqst = {
             "rectype": "atl06rec.elevation",
         }
         d = sliderule.source("definition", rqst)
         assert d["delta_time"]["offset"] == 192
 
-    def test_version(self):
-        icesat2.init(SERVER)
+    def test_version(self, server):
+        icesat2.init(server)
         rsps = sliderule.source("version", {})
         assert 'server' in rsps
         assert 'version' in rsps['server']

tests/test_icesat2.py

@@ -8,7 +8,6 @@
 import os.path
 
 TESTDIR = Path(__file__).parent
-SERVER = "icesat2sliderule.org"
 
 # Change connection timeout from default 10s to 1s
 sliderule.set_rqst_timeout((1, 60))
@@ -43,14 +42,14 @@ def test_init_badurl(self):
         with pytest.raises( (ConnectTimeout, ConnectionError) ):
             icesat2.init('incorrect.org')
 
-    def test_get_version(self):
-        icesat2.init(SERVER)
+    def test_get_version(self, server):
+        icesat2.init(server)
         version = icesat2.get_version()
         assert isinstance(version, dict)
         assert {'icesat2', 'server', 'client'} <= version.keys()
 
-    def test_cmr(self, grandmesa):
-        icesat2.init(SERVER)
+    def test_cmr(self, grandmesa, server):
+        icesat2.init(server)
         granules = icesat2.cmr(polygon=grandmesa,
             time_start='2018-10-01',
             time_end='2018-12-01')