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)
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)
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)
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)
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)
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)
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)