Add test for get_scalar_constant

Author: shreyas3156

tests/tensor/test_basic.py

@@ -51,6 +51,7 @@
     flatnonzero,
     flatten,
     full_like,
+    get_scalar_constant,
     get_underlying_scalar_constant,
     get_vector_length,
     horizontal_stack,
@@ -3429,6 +3430,12 @@ def test_None_and_NoneConst(self, only_process_constants):
         )
 
 
+def test_get_scalar_constant():
+    with pytest.raises(NotScalarConstantError):
+        get_scalar_constant(np.zeros(5))
+    assert get_scalar_constant(np.array(4)) == 4
+
+
 def test_complex_mod_failure():
     # Make sure % fails on complex numbers.
     x = vector(dtype="complex64")