def test_repair_image_nonfinite(): for i in range(0, 9): for non_number in [np.nan, np.inf]: a = np.ones((9)) a[i] = non_number b = a.reshape(3, 3) c = repair_image_nonfinite(b) assert(np.isfinite(c).all())
def test_repair_image_nonfinite(): for i in range(0, 9): for non_number in [np.nan, np.inf]: a = np.ones(9) a[i] = non_number b = a.reshape(3, 3) with pytest.warns(SunpyDeprecationWarning): c = repair_image_nonfinite(b) assert (np.isfinite(c).all())