TEST: Use pytest-httpserver to test validate_from_url

Author: effigies

nibabel/tests/test_image_api.py

@@ -572,6 +572,28 @@ def validate_to_from_bytes(self, imaker, params):
                 del img_a
                 del img_b
 
+    @pytest.fixture(autouse=True)
+    def setup(self, httpserver):
+        """Make pytest fixtures available to validate functions"""
+        self.httpserver = httpserver
+
+    def validate_from_url(self, imaker, params):
+        server = self.httpserver
+
+        img = imaker()
+        img_bytes = img.to_bytes()
+
+        server.expect_oneshot_request("/img").respond_with_data(img_bytes)
+        url = server.url_for("/img")
+        assert url.startswith("http://")  # Check we'll trigger an HTTP handler
+        rt_img = img.__class__.from_url(url)
+
+        assert rt_img.to_bytes() == img_bytes
+        assert self._header_eq(img.header, rt_img.header)
+        assert np.array_equal(img.get_fdata(), rt_img.get_fdata())
+        del img
+        del rt_img
+
     @staticmethod
     def _header_eq(header_a, header_b):
         """ Header equality check that can be overridden by a subclass of this test
@@ -583,7 +605,6 @@ def _header_eq(header_a, header_b):
         return header_a == header_b
 
 
-
 class LoadImageAPI(GenericImageAPI,
                    DataInterfaceMixin,
                    AffineMixin,

setup.cfg

@@ -61,6 +61,7 @@ test =
     pytest !=5.3.4
     pytest-cov
     pytest-doctestplus
+    pytest-httpserver
 zstd =
     pyzstd >= 0.14.3
 all =