tests : add more tensor tests (python)

Author: ManifoldFR

tests/test_eigen_tensor.cpp

@@ -28,10 +28,10 @@ NB_MODULE(test_eigen_tensor_ext, m) {
         return a + b;
     }, "a"_a, "b"_a);
     m.def("add3dTensor_nc", [](const Tensor3d &a, const Tensor3d &b) -> Tensor3d {
-        return (a + b).eval();
+        return a + b;
     }, "a"_a.noconvert(), "b"_a.noconvert());
     m.def("square3dTensorR", [](const RowTensor3d &a) -> RowTensor3d {
-        return a.square().eval();
+        return a.square();
     }, "a"_a.noconvert());
 
     m.def("mul3dTensor", [](double a, const Tensor3d &b) -> Tensor3d {

tests/test_eigen_tensor.py

@@ -19,3 +19,26 @@ def test01_tensor3d():
     a_colmaj = np.asfortranarray(a)
     with pytest.raises(TypeError, match='incompatible function arguments'):
         t.square3dTensorR(a_colmaj)
+
+    # add3dTensor: col-major inputs (with implicit conversion from row-major)
+    b = np.arange(12, 24, dtype=float).reshape(2, 3, 2)
+    assert_array_equal(t.add3dTensor(a, b), a + b)
+    # row-major numpy arrays are implicitly converted
+    assert_array_equal(t.add3dTensor(a, a), a + a)
+
+    # add3dTensor_nc: noconvert — rejects row-major (C-order) arrays
+    with pytest.raises(TypeError, match='incompatible function arguments'):
+        t.add3dTensor_nc(a, b)
+    # Fortran-order (col-major) arrays are accepted without conversion
+    assert_array_equal(t.add3dTensor_nc(a_colmaj, np.asfortranarray(b)), a + b)
+
+    # mul3dTensor: scalar * col-major tensor
+    assert_array_equal(t.mul3dTensor(2.0, a), 2.0 * a)
+
+    # mul3dTensorMap: read-only TensorMap, returns a new tensor
+    assert_array_equal(t.mul3dTensorMap(3.0, a_colmaj), 3.0 * a)
+
+    # mul3dTensorMapInPlace: mutates the array in place via TensorMap
+    arr = np.asfortranarray(a.copy())
+    t.mul3dTensorMapInPlace(2.0, arr)
+    assert_array_equal(arr, 2.0 * a)