Expand test coverage

helmholtz-analytics · Aug 23, 2021 · be8979d · be8979d
1 parent f1cadcb
commit be8979d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/heat/core/tests/test_manipulations.py b/heat/core/tests/test_manipulations.py
@@ -3276,14 +3276,22 @@ def test_tile(self):
         self.assertTrue((np_tiled == ht_tiled.numpy()).all())
         self.assertTrue(ht_tiled.dtype is x.dtype)
 
-        # test scalar x
+        # test scalar DNDarray x
         x = ht.array(9.0)
         reps = (2, 1)
         ht_tiled = ht.tile(x, reps)
         np_tiled = np.tile(x.numpy(), reps)
         self.assertTrue((np_tiled == ht_tiled.numpy()).all())
         self.assertTrue(ht_tiled.dtype is x.dtype)
 
+        # test scalar x
+        x = 10
+        reps = (2, 1)
+        ht_tiled = ht.tile(x, reps)
+        np_tiled = np.tile(x, reps)
+        self.assertTrue((np_tiled == ht_tiled.numpy()).all())
+        self.assertTrue(ht_tiled.dtype is ht.int64)
+
         # test distributed tile along split axis
         # len(reps) > x.ndim
         split = 1