Use the promote_dtypes helper instead of _array_module.result_type

Author: asmeurer

array_api_tests/test_linalg.py

@@ -17,7 +17,7 @@
 from hypothesis.strategies import booleans, composite, none, integers, shared
 
 from .array_helpers import (assert_exactly_equal, ndindex, asarray,
-                            numeric_dtype_objects)
+                            numeric_dtype_objects, promote_dtypes)
 from .hypothesis_helpers import (xps, dtypes, shapes, kwargs, matrix_shapes,
                                  square_matrix_shapes, symmetric_matrices,
                                  positive_definite_matrices, MAX_ARRAY_SIZE,
@@ -132,8 +132,7 @@ def test_cross(x1_x2_kw):
 
     res = linalg.cross(x1, x2, **kw)
 
-    # TODO: Replace result_type() with a helper function
-    assert res.dtype == _array_module.result_type(x1, x2), "cross() did not return the correct dtype"
+    assert res.dtype == promote_dtypes(x1, x2), "cross() did not return the correct dtype"
     assert res.shape == shape, "cross() did not return the correct shape"
 
     # cross is too different from other functions to use _test_stacks, and it
@@ -269,8 +268,7 @@ def test_matmul(x1, x2):
     else:
         res = linalg.matmul(x1, x2)
 
-    # TODO: Replace result_type() with a helper function
-    assert res.dtype == _array_module.result_type(x1, x2), "matmul() did not return the correct dtype"
+    assert res.dtype == promote_dtypes(x1, x2), "matmul() did not return the correct dtype"
 
     if len(x1.shape) == len(x2.shape) == 1:
         assert res.shape == ()