Exemplo n.º 1
0
def test_check_1d_on_1d():
    assert check_1d([1, 2, 3]) == [1, 2, 3]
Exemplo n.º 2
0
def test_check_1d_on_2d():
    with pytest.raises(ValueError):
        check_1d(np.arange(10).reshape(-1, 2))
Exemplo n.º 3
0
def test_check_1d_on_dirty_1d():
    assert check_1d([1, np.array(2), 3]) == [1, 2, 3]
Exemplo n.º 4
0
def test_check_1d_on_1d():
    assert check_1d([1, 2, 3]) == [1, 2, 3]
Exemplo n.º 5
0
def test_check_1d_on_2d():
    with pytest.raises(ValueError):
        check_1d(np.arange(10).reshape(-1, 2))
Exemplo n.º 6
0
def test_check_1d_on_dirty_1d():
    assert check_1d([1, np.array(2), 3]) == [1, 2, 3]