def test_box_nan(data): out = box(data, 1, 1, nan=True) np.testing.assert_array_equal(out, [ [0, 1, 2], [5, np.nan, 7], [10, 11, 12] ])
def test_box(data): out = box(data, 1, 1) np.testing.assert_array_equal(out, [ [0, 1, 2], [5, 6, 7], [10, 11, 12] ])
def test_box_size(data): out = box(data, 2, 2, size=2) np.testing.assert_array_equal(out, [ [0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19], [20, 21, 22, 23, 24] ])
def test_box_borders(data): out_i_0 = box(data, 0, 1) np.testing.assert_array_equal(out_i_0, [ [0, 1], [5, 6], [10, 11] ]) out_i_nx = box(data, 4, 1) np.testing.assert_array_equal(out_i_nx, [ [3, 4], [8, 9], [13, 14] ]) out_j_0 = box(data, 1, 0) np.testing.assert_array_equal(out_j_0, [ [0, 1, 2], [5, 6, 7] ]) out_j_ny = box(data, 1, 5) np.testing.assert_array_equal(out_j_ny, [ [20, 21, 22], [25, 26, 27] ])
def test_box_err(data): with pytest.raises(TypeError): box(0, 1, 1) with pytest.raises(ValueError): box(data, -1, 1) with pytest.raises(ValueError): box(data, 1, -1) with pytest.raises(ValueError): box(data, 5, 1) with pytest.raises(ValueError): box(data, 1, 6) with pytest.raises(TypeError): box(data, 1, 1, size=1.) with pytest.raises(ValueError): box(data, 1, 1, size=-1)