fix test

Signed-off-by: Pradyot Ranjan <99216956+pradyotRanjan@users.noreply.github.com>

Author: prady0t

tests/test_funcs.py

@@ -1887,8 +1887,8 @@ def test_device(self, xp: ModuleType, device: Device):
 class TestAngle:
     def test_simple(self, xp: ModuleType):
         a = xp.asarray([1, 0])
-        expected = xp.asarray([0.0, 0.0], dtype=xp.float64)
         res = angle(a)
+        expected = xp.asarray([0.0, 0.0], dtype=res.dtype)
         xp_assert_equal(res, expected)
 
     def test_basic(self, xp: ModuleType):