@@ -146,7 +146,7 @@ def test_get_structure_plot_params():
146146 pp = SCENE_FULL ._get_structure_eps_plot_params (
147147 medium = SCENE_FULL .medium , freq = 1 , eps_min = 1 , eps_max = 2
148148 )
149- expected_color = mpl . cm .get_cmap (STRUCTURE_EPS_CMAP )(0.0 )
149+ expected_color = plt .get_cmap (STRUCTURE_EPS_CMAP )(0.0 )
150150 assert np .allclose (pp .facecolor , expected_color )
151151 pp = SCENE_FULL ._get_structure_eps_plot_params (medium = td .PEC , freq = 1 , eps_min = 1 , eps_max = 2 )
152152 assert pp .facecolor == "gold"
@@ -165,7 +165,7 @@ def test_structure_eps_color_mapping():
165165 norm = norm ,
166166 reverse = False ,
167167 )
168- expected_min = mpl . cm .get_cmap (STRUCTURE_EPS_CMAP )(norm (1.0 ))
168+ expected_min = plt .get_cmap (STRUCTURE_EPS_CMAP )(norm (1.0 ))
169169 assert np .allclose (pp_min .facecolor , expected_min )
170170
171171 pp_max = SCENE_FULL ._get_structure_eps_plot_params (
@@ -176,7 +176,7 @@ def test_structure_eps_color_mapping():
176176 norm = norm ,
177177 reverse = False ,
178178 )
179- expected_max = mpl . cm .get_cmap (STRUCTURE_EPS_CMAP )(norm (5.0 ))
179+ expected_max = plt .get_cmap (STRUCTURE_EPS_CMAP )(norm (5.0 ))
180180 assert np .allclose (pp_max .facecolor , expected_max )
181181
182182 pp_min_reverse = SCENE_FULL ._get_structure_eps_plot_params (
@@ -187,7 +187,7 @@ def test_structure_eps_color_mapping():
187187 norm = norm ,
188188 reverse = True ,
189189 )
190- expected_min_reverse = mpl . cm .get_cmap (STRUCTURE_EPS_CMAP_R )(norm (1.0 ))
190+ expected_min_reverse = plt .get_cmap (STRUCTURE_EPS_CMAP_R )(norm (1.0 ))
191191 assert np .allclose (pp_min_reverse .facecolor , expected_min_reverse )
192192
193193 pp_max_reverse = SCENE_FULL ._get_structure_eps_plot_params (
@@ -198,7 +198,7 @@ def test_structure_eps_color_mapping():
198198 norm = norm ,
199199 reverse = True ,
200200 )
201- expected_max_reverse = mpl . cm .get_cmap (STRUCTURE_EPS_CMAP_R )(norm (5.0 ))
201+ expected_max_reverse = plt .get_cmap (STRUCTURE_EPS_CMAP_R )(norm (5.0 ))
202202 assert np .allclose (pp_max_reverse .facecolor , expected_max_reverse )
203203
204204
0 commit comments