Arithmetic Operation Testing

tests/test_arithmetic.py

@@ -0,0 +1,152 @@
+import random
+
+import pytest
+
+import arrayfire_wrapper.dtypes as dtype
+import arrayfire_wrapper.lib as wrapper
+from tests.utility_functions import check_type_supported, get_all_types
+
+
+@pytest.mark.parametrize(
+    "shape",
+    [
+        (),
+        (random.randint(1, 10),),
+        (random.randint(1, 10), random.randint(1, 10)),
+        (random.randint(1, 10), random.randint(1, 10), random.randint(1, 10)),
+        (random.randint(1, 10), random.randint(1, 10), random.randint(1, 10), random.randint(1, 10)),
+    ],
+)
+def test_add_shapes(shape: tuple) -> None:
+    """Test addition operation between two arrays of the same shape"""
+    lhs = wrapper.randu(shape, dtype.f16)
+    rhs = wrapper.randu(shape, dtype.f16)
+
+    result = wrapper.add(lhs, rhs)
+
+    assert wrapper.get_dims(result)[0 : len(shape)] == shape  # noqa: E203, W291
+
+
+def test_add_different_shapes() -> None:
+    """Test if addition handles arrays of different shapes"""
+    with pytest.raises(RuntimeError):
+        lhs_shape = (2, 3)
+        rhs_shape = (3, 2)
+        dtypes = dtype.f16
+        lhs = wrapper.randu(lhs_shape, dtypes)
+        rhs = wrapper.randu(rhs_shape, dtypes)
+
+        wrapper.add(lhs, rhs)
+
+
+@pytest.mark.parametrize("dtype_name", get_all_types())
+def test_add_supported_dtypes(dtype_name: dtype.Dtype) -> None:
+    """Test addition operation across all supported data types."""
+    check_type_supported(dtype_name)
+    shape = (5, 5)  # Using a common shape for simplicity
+    lhs = wrapper.randu(shape, dtype_name)
+    rhs = wrapper.randu(shape, dtype_name)
+    result = wrapper.add(lhs, rhs)
+    assert dtype.c_api_value_to_dtype(wrapper.get_type(result)) == dtype_name, f"Failed for dtype: {dtype_name}"
+
+
+@pytest.mark.parametrize(
+    "invdtypes",
+    [
+        dtype.c64,
+        dtype.f64,
+    ],
+)
+def test_add_unsupported_dtypes(invdtypes: dtype.Dtype) -> None:
+    """Test addition operation across all supported data types."""
+    with pytest.raises(RuntimeError):
+        shape = (5, 5)
+        lhs = wrapper.randu(shape, invdtypes)
+        rhs = wrapper.randu(shape, invdtypes)
+        result = wrapper.add(lhs, rhs)
+        assert dtype.c_api_value_to_dtype(wrapper.get_type(result)) == invdtypes, f"Didn't Fail for Dtype: {invdtypes}"
+
+
+def test_add_zero_sized_arrays() -> None:
+    """Test addition with arrays where at least one array has zero size."""
+    with pytest.raises(RuntimeError):
+        zero_shape = (0, 5)
+        normal_shape = (5, 5)
+        zero_array = wrapper.randu(zero_shape, dtype.f32)
+        normal_array = wrapper.randu(normal_shape, dtype.f32)
+
+        # Test addition when lhs is zero-sized
+        result_lhs_zero = wrapper.add(zero_array, normal_array)
+        assert wrapper.get_dims(result_lhs_zero) == zero_shape
+
+
+@pytest.mark.parametrize(
+    "shape",
+    [
+        (),
+        (random.randint(1, 10),),
+        (random.randint(1, 10), random.randint(1, 10)),
+        (random.randint(1, 10), random.randint(1, 10), random.randint(1, 10)),
+        (random.randint(1, 10), random.randint(1, 10), random.randint(1, 10), random.randint(1, 10)),
+    ],
+)
+def test_subtract_shapes(shape: tuple) -> None:
+    """Test subtraction operation between two arrays of the same shape"""
+    lhs = wrapper.randu(shape, dtype.f16)
+    rhs = wrapper.randu(shape, dtype.f16)
+
+    result = wrapper.sub(lhs, rhs)
+
+    assert wrapper.get_dims(result)[0 : len(shape)] == shape  # noqa: E203, W291
+
+
+def test_subtract_different_shapes() -> None:
+    """Test if subtraction handles arrays of different shapes"""
+    with pytest.raises(RuntimeError):
+        lhs_shape = (2, 3)
+        rhs_shape = (3, 2)
+        dtypes = dtype.f16
+        lhs = wrapper.randu(lhs_shape, dtypes)
+        rhs = wrapper.randu(rhs_shape, dtypes)
+
+        wrapper.sub(lhs, rhs)
+
+
+@pytest.mark.parametrize("dtype_name", get_all_types())
+def test_subtract_supported_dtypes(dtype_name: dtype.Dtype) -> None:
+    """Test subtraction operation across all supported data types."""
+    check_type_supported(dtype_name)
+    shape = (5, 5)
+    lhs = wrapper.randu(shape, dtype_name)
+    rhs = wrapper.randu(shape, dtype_name)
+    result = wrapper.sub(lhs, rhs)
+    assert dtype.c_api_value_to_dtype(wrapper.get_type(result)) == dtype_name, f"Failed for dtype: {dtype_name}"
+
+
+@pytest.mark.parametrize(
+    "invdtypes",
+    [
+        dtype.c64,
+        dtype.f64,
+    ],
+)
+def test_subtract_unsupported_dtypes(invdtypes: dtype.Dtype) -> None:
+    """Test subtraction operation for unsupported data types."""
+    with pytest.raises(RuntimeError):
+        shape = (5, 5)
+        lhs = wrapper.randu(shape, invdtypes)
+        rhs = wrapper.randu(shape, invdtypes)
+        result = wrapper.sub(lhs, rhs)
+        assert result == invdtypes, f"Didn't Fail for Dtype: {invdtypes}"
+
+
+def test_subtract_zero_sized_arrays() -> None:
+    """Test subtraction with arrays where at least one array has zero size."""
+    with pytest.raises(RuntimeError):
+        zero_shape = (0, 5)
+        normal_shape = (5, 5)
+        zero_array = wrapper.randu(zero_shape, dtype.f32)
+        normal_array = wrapper.randu(normal_shape, dtype.f32)
+
+        result_lhs_zero = wrapper.sub(zero_array, normal_array)
+        assert wrapper.get_dims(result_lhs_zero) == zero_shape