def test_mask_interpolate(): # A coordinate array with coordinates of two points almost coinciding. coords = np.array([[0, 0], [1e-7, 1e-7], [1, 1], [1, 0]]) with warnings.catch_warnings(record=True) as w: warnings.simplefilter("always") plotter.mask_interpolate(coords, np.ones(len(coords)), a=1) assert len(w) == 1 assert issubclass(w[-1].category, RuntimeWarning) assert "coinciding" in str(w[-1].message) with warnings.catch_warnings(): warnings.simplefilter("ignore") pytest.raises(ValueError, plotter.mask_interpolate, coords, np.ones(len(coords))) pytest.raises(ValueError, plotter.mask_interpolate, coords, np.ones(2 * len(coords)))
def test_mask_interpolate(): # A coordinate array with coordinates of two points almost coinciding. coords = np.random.rand(10, 2) coords[5] *= 1e-8 coords[5] += coords[0] warnings.simplefilter("ignore") with warnings.catch_warnings(record=True) as w: warnings.simplefilter("always") plotter.mask_interpolate(coords, np.ones(len(coords)), a=1) assert len(w) == 1 assert issubclass(w[-1].category, RuntimeWarning) assert "coinciding" in str(w[-1].message) assert_raises(ValueError, plotter.mask_interpolate, coords, np.ones(len(coords))) assert_raises(ValueError, plotter.mask_interpolate, coords, np.ones(2 * len(coords)))