add scalar with k test

cupy · Mar 22, 2018 · cf6547b · cf6547b
1 parent ab7bf00
commit cf6547b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/cupy_tests/creation_tests/test_matrix.py b/tests/cupy_tests/creation_tests/test_matrix.py
@@ -41,3 +41,11 @@ def test_diagflat3(self, xp):
     @testing.numpy_cupy_array_equal()
     def test_diagflat_from_scalar(self, xp):
         return xp.diagflat(3)
+
+    @testing.numpy_cupy_array_equal()
+    def test_diagflat_from_scalar_with_k0(self, xp):
+        return xp.diagflat(3, 0)
+
+    @testing.numpy_cupy_array_equal()
+    def test_diagflat_from_scalar_with_k1(self, xp):
+        return xp.diagflat(3, 1)