Example #1
0
def test_datashape_measure():
    assert_dshape_equal(dshape('int'), dshape('int'))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape('int'), dshape('string'))
    assert 'int32 != string' in str(e.value)
    assert '_.measure' in str(e.value)
Example #2
0
def test_datashape_measure():
    assert_dshape_equal(dshape('int'), dshape('int'))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape('int'), dshape('string'))
    assert 'int32 != string' in str(e.value)
    assert '_.measure' in str(e.value)
Example #3
0
def test_nested():
    assert_dshape_equal(dshape("var * {a: 3 * {b: int32}}"), dshape("var * {a: 3 * {b: int32}}"))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape("var * {a: 3 * {b: int32}}"), dshape("var * {a: 3 * {b: float32}}"))
    assert "'int32' != 'float32'" in str(e.value)
    assert "_.measure['a'].measure['b'].name" in str(e.value)
Example #4
0
def test_datashape_measure():
    assert_dshape_equal(dshape("int"), dshape("int"))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape("int"), dshape("string"))
    assert "int32 != string" in str(e.value)
    assert "_.measure" in str(e.value)
Example #5
0
def test_function(dshape_, contains):
    base = dshape('(int32, int64) -> int64')
    assert_dshape_equal(base, base)

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape(dshape_), base)
    for c in contains:
        assert c in str(e.value)
Example #6
0
def test_function(dshape_, contains):
    base = dshape('(int32, int64) -> int64')
    assert_dshape_equal(base, base)

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape(dshape_), base)
    for c in contains:
        assert c in str(e.value)
Example #7
0
def test_nested():
    assert_dshape_equal(
        dshape('var * {a: 3 * {b: int32}}'),
        dshape('var * {a: 3 * {b: int32}}'),
    )

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(
            dshape('var * {a: 3 * {b: int32}}'),
            dshape('var * {a: 3 * {b: float32}}'),
        )
    assert "'int32' != 'float32'" in str(e.value)
    assert "_.measure['a'].measure['b'].name" in str(e.value)
Example #8
0
def test_dim():
    assert_dshape_equal(dshape('var * int'), dshape('var * int'))
    assert_dshape_equal(dshape('3 * string'), dshape('3 * string'))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape('var * int'), dshape('3 * int'))
    assert 'var != 3' in str(e.value)
    assert '_.shape[0]' in str(e.value)

    assert_dshape_equal(dshape('var * var * int'), dshape('var * var * int'))
    assert_dshape_equal(dshape('var * 3 * string'), dshape('var * 3 * string'))
    assert_dshape_equal(
        dshape('3 * var * float32'),
        dshape('3 * var * float32'),
    )
    assert_dshape_equal(
        dshape('3 * 3 * datetime'),
        dshape('3 * 3 * datetime'),
    )

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(
            dshape('var * var * int'),
            dshape('3 * var * int'),
        )
    assert 'var != 3' in str(e.value)
    assert '_.shape[0]' in str(e.value)

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(
            dshape('var * var * int'),
            dshape('var * 3 * int'),
        )
    assert 'var != 3' in str(e.value)
    assert '_.shape[1]' in str(e.value)
Example #9
0
def test_dim():
    assert_dshape_equal(dshape('var * int'), dshape('var * int'))
    assert_dshape_equal(dshape('3 * string'), dshape('3 * string'))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape('var * int'), dshape('3 * int'))
    assert 'var != 3' in str(e.value)
    assert '_.shape[0]' in str(e.value)

    assert_dshape_equal(dshape('var * var * int'), dshape('var * var * int'))
    assert_dshape_equal(dshape('var * 3 * string'), dshape('var * 3 * string'))
    assert_dshape_equal(
        dshape('3 * var * float32'),
        dshape('3 * var * float32'),
    )
    assert_dshape_equal(
        dshape('3 * 3 * datetime'),
        dshape('3 * 3 * datetime'),
    )

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(
            dshape('var * var * int'),
            dshape('3 * var * int'),
        )
    assert 'var != 3' in str(e.value)
    assert '_.shape[0]' in str(e.value)

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(
            dshape('var * var * int'),
            dshape('var * 3 * int'),
        )
    assert 'var != 3' in str(e.value)
    assert '_.shape[1]' in str(e.value)
Example #10
0
def test_dim():
    assert_dshape_equal(dshape("var * int"), dshape("var * int"))
    assert_dshape_equal(dshape("3 * string"), dshape("3 * string"))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape("var * int"), dshape("3 * int"))
    assert "var != 3" in str(e.value)
    assert "_.shape[0]" in str(e.value)

    assert_dshape_equal(dshape("var * var * int"), dshape("var * var * int"))
    assert_dshape_equal(dshape("var * 3 * string"), dshape("var * 3 * string"))
    assert_dshape_equal(dshape("3 * var * float32"), dshape("3 * var * float32"))
    assert_dshape_equal(dshape("3 * 3 * datetime"), dshape("3 * 3 * datetime"))

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape("var * var * int"), dshape("3 * var * int"))
    assert "var != 3" in str(e.value)
    assert "_.shape[0]" in str(e.value)

    with pytest.raises(AssertionError) as e:
        assert_dshape_equal(dshape("var * var * int"), dshape("var * 3 * int"))
    assert "var != 3" in str(e.value)
    assert "_.shape[1]" in str(e.value)