merge

briangu · Dec 13, 2023 · 5810cf4 · 5810cf4
1 parent c181662
commit 5810cf4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 18 deletions.
diff --git a/tests/kgtests/interop/test_vectorization.kg b/tests/kgtests/interop/test_vectorization.kg
@@ -0,0 +1,4 @@
+t("{2*x}'!1000"; {2*x}'!1000; 2*!1000)
+
+a::!1000
+t("a::!1000;2*a"; 2*a; 2*!1000)
diff --git a/tests/test_extra_suite.py b/tests/test_extra_suite.py
@@ -19,6 +19,11 @@ def test_extra_spaces(self):
         r = klong("a::{ 1 + 1 };a()")
         self.assertEqual(r, 2)
 
+    def test_vectorized(self):
+        klong = KlongInterpreter()
+        r = klong("2*!1000")
+        self.assertTrue(kg_equal(r, np.arange(1000)*2))
+
     def test_power_type(self):
         klong = KlongInterpreter()
         r = klong("2^3")

diff --git a/tests/test_vectorization.py b/tests/test_vectorization.py