Type data return types

Author: dstansby

src/napari_matplotlib/tests/conftest.py

@@ -1,9 +1,11 @@
 import os
 
+from typing import Dict, Any
 import numpy as np
 import pytest
 from skimage import data
 
+DataMeta = Dict[np.ndarray, Dict[str, Any]]
 
 @pytest.fixture(
     params=[
@@ -12,17 +14,17 @@
         ((100, 100, 3), {"rgb": True}),
     ]
 )
-def image_data(request):
+def image_data(request) -> DataMeta:
     return np.ones(request.param[0]), request.param[1]
 
 
 @pytest.fixture
-def astronaut_data():
+def astronaut_data() -> DataMeta:
     return data.astronaut(), {"rgb": True}
 
 
 @pytest.fixture
-def brain_data():
+def brain_data() -> DataMeta:
     return data.brain(), {"rgb": False}