dump

honno · honno · commit 1b97e588dc05 · 2024-04-30T18:15:23.000+01:00
diff --git a/array_api_tests/test_inspection_functions.py b/array_api_tests/test_inspection_functions.py
@@ -0,0 +1,6 @@
+from . import xp
+
+
+def test_array_namespace_info():
+    assert hasattr(xp, "__array_namespace_info__")
+    # TODO: test output
diff --git a/array_api_tests/test_operators_and_elementwise_functions.py b/array_api_tests/test_operators_and_elementwise_functions.py
@@ -933,6 +933,16 @@ def test_ceil(x):
     unary_assert_against_refimpl("ceil", x, out, math.ceil, strict_check=True)
 
 
+@pytest.mark.min_version("2023.12")
+@given(hh.arrays(dtype=xps.floating_dtypes(), shape=hh.shapes()))
+def test_clip(x):
+    # TODO: test min/max kwargs, adjust values testing accordingly
+    out = xp.clip(x)
+    ph.assert_dtype("clip", in_dtype=x.dtype, out_dtype=out.dtype)
+    ph.assert_shape("clip", out_shape=out.shape, expected=x.shape)
+    ph.assert_array_elements("clip", out=out, expected=x)
+
+
 if api_version >= "2022.12":
 
     @given(hh.arrays(dtype=xps.complex_dtypes(), shape=hh.shapes()))
@@ -943,6 +953,15 @@ def test_conj(x):
         unary_assert_against_refimpl("conj", x, out, operator.methodcaller("conjugate"))
 
 
+@pytest.mark.min_version("2023.12")
+@given(*hh.two_mutual_arrays(dh.real_float_dtypes))
+def test_copysign(x1, x2):
+    out = xp.copysign(x1, x2)
+    ph.assert_dtype("copysign", in_dtype=[x1.dtype, x2.dtype], out_dtype=out.dtype)
+    ph.assert_result_shape("copysign", in_shapes=[x1.shape, x2.shape], out_shape=out.shape)
+    # TODO: values testing
+
+
 @given(hh.arrays(dtype=hh.all_floating_dtypes(), shape=hh.shapes()))
 def test_cos(x):
     out = xp.cos(x)
@@ -1095,6 +1114,15 @@ def test_greater_equal(ctx, data):
     )
 
 
+@pytest.mark.min_version("2023.12")
+@given(*hh.two_mutual_arrays(dh.real_float_dtypes))
+def test_hypot(x1, x2):
+    out = xp.hypot(x1, x2)
+    ph.assert_dtype("hypot", in_dtype=[x1.dtype, x2.dtype], out_dtype=out.dtype)
+    ph.assert_result_shape("hypot", in_shapes=[x1.shape, x2.shape], out_shape=out.shape)
+    binary_assert_against_refimpl("hypot", x1, x2, out, math.hypot)
+
+
 if api_version >= "2022.12":
 
     @given(hh.arrays(dtype=xps.complex_dtypes(), shape=hh.shapes()))
@@ -1261,6 +1289,24 @@ def test_logical_xor(x1, x2):
     )
 
 
+@pytest.mark.min_version("2023.12")
+@given(*hh.two_mutual_arrays(dh.real_float_dtypes))
+def test_maximum(x1, x2):
+    out = xp.maximum(x1, x2)
+    ph.assert_dtype("maximum", in_dtype=[x1.dtype, x2.dtype], out_dtype=out.dtype)
+    ph.assert_result_shape("maximum", in_shapes=[x1.shape, x2.shape], out_shape=out.shape)
+    binary_assert_against_refimpl("maximum", x1, x2, out, max, strict_check=True)
+
+
+@pytest.mark.min_version("2023.12")
+@given(*hh.two_mutual_arrays(dh.real_float_dtypes))
+def test_minimum(x1, x2):
+    out = xp.minimum(x1, x2)
+    ph.assert_dtype("minimum", in_dtype=[x1.dtype, x2.dtype], out_dtype=out.dtype)
+    ph.assert_result_shape("minimum", in_shapes=[x1.shape, x2.shape], out_shape=out.shape)
+    binary_assert_against_refimpl("minimum", x1, x2, out, min, strict_check=True)
+
+
 @pytest.mark.parametrize("ctx", make_binary_params("multiply", dh.numeric_dtypes))
 @given(data=st.data())
 def test_multiply(ctx, data):
diff --git a/array_api_tests/test_statistical_functions.py b/array_api_tests/test_statistical_functions.py
@@ -16,6 +16,15 @@
 from .typing import DataType
 
 
+@given(hh.arrays(dtype=xps.numeric_dtypes(), shape=hh.shapes(min_dims=1, max_dims=1)))
+def test_cumulative_sum(x):
+    # TODO: test kwargs + diff shapes, adjust shape and values testing accordingly
+    out = xp.cumulative_sum(x)
+    # TODO: assert dtype
+    ph.assert_shape("cumulative_sum", out_shape=out.shape, expected=x.shape)
+    # TODO: assert values
+
+
 def kwarg_dtypes(dtype: DataType) -> st.SearchStrategy[Optional[DataType]]:
     dtypes = [d2 for d1, d2 in dh.promotion_table if d1 == dtype]
     dtypes = [d for d in dtypes if not isinstance(d, _UndefinedStub)]
