def test_jax_shape_ops():
x_np = np.zeros((20, 3))
x = Shape()(aet.as_tensor_variable(x_np))
x_fg = FunctionGraph([], [x])
compare_jax_and_py(x_fg, [], must_be_device_array=False)
x = Shape_i(1)(aet.as_tensor_variable(x_np))
x_fg = FunctionGraph([], [x])
compare_jax_and_py(x_fg, [], must_be_device_array=False)
def test_jax_shape_ops():
x_np = np.zeros((20, 3))
x = Shape()(aet.as_tensor_variable(x_np))
x_fg = FunctionGraph([], [x])
compare_jax_and_py(x_fg, [], must_be_device_array=False)
x = Shape_i(1)(aet.as_tensor_variable(x_np))
x_fg = FunctionGraph([], [x])
compare_jax_and_py(x_fg, [], must_be_device_array=False)
x = SpecifyShape()(aet.as_tensor_variable(x_np), (20, 3))
x_fg = FunctionGraph([], [x])
compare_jax_and_py(x_fg, [])
with config.change_flags(compute_test_value="off"):
x = SpecifyShape()(aet.as_tensor_variable(x_np), (2, 3))
x_fg = FunctionGraph([], [x])
with pytest.raises(AssertionError):
compare_jax_and_py(x_fg, [])
def test_shape_i_hash():
assert isinstance(Shape_i(np.int64(1)).__hash__(), int)
def test_shape_i_basics():
with pytest.raises(TypeError):
Shape_i(0)([1, 2])
with pytest.raises(TypeError):
Shape_i(0)(scalar())