|
2 | 2 |
|
3 | 3 | import numpy as np |
4 | 4 | import cv2 |
5 | | -from skimage import data |
| 5 | +import pytest |
| 6 | + |
| 7 | +skimage_data = pytest.importorskip("skimage.data") |
6 | 8 |
|
7 | 9 | import rlemasklib |
8 | 10 |
|
@@ -34,17 +36,17 @@ def test_rectangle(self): |
34 | 36 | self._compare_moments(mask) |
35 | 37 |
|
36 | 38 | def test_coins_thresholded(self): |
37 | | - coins = data.coins() |
| 39 | + coins = skimage_data.coins() |
38 | 40 | mask = (coins > 100).astype(np.uint8) |
39 | 41 | self._compare_moments(mask) |
40 | 42 |
|
41 | 43 | def test_camera_thresholded(self): |
42 | | - camera = data.camera() |
| 44 | + camera = skimage_data.camera() |
43 | 45 | mask = (camera > 128).astype(np.uint8) |
44 | 46 | self._compare_moments(mask) |
45 | 47 |
|
46 | 48 | def test_text_thresholded(self): |
47 | | - text = data.text() |
| 49 | + text = skimage_data.text() |
48 | 50 | mask = (text < 100).astype(np.uint8) |
49 | 51 | self._compare_moments(mask) |
50 | 52 |
|
@@ -99,17 +101,17 @@ def test_rectangle(self): |
99 | 101 | self._compare_hu_moments(mask) |
100 | 102 |
|
101 | 103 | def test_coins_thresholded(self): |
102 | | - coins = data.coins() |
| 104 | + coins = skimage_data.coins() |
103 | 105 | mask = (coins > 100).astype(np.uint8) |
104 | 106 | self._compare_hu_moments(mask) |
105 | 107 |
|
106 | 108 | def test_camera_thresholded(self): |
107 | | - camera = data.camera() |
| 109 | + camera = skimage_data.camera() |
108 | 110 | mask = (camera > 128).astype(np.uint8) |
109 | 111 | self._compare_hu_moments(mask) |
110 | 112 |
|
111 | 113 | def test_text_thresholded(self): |
112 | | - text = data.text() |
| 114 | + text = skimage_data.text() |
113 | 115 | mask = (text < 100).astype(np.uint8) |
114 | 116 | self._compare_hu_moments(mask) |
115 | 117 |
|
|
0 commit comments