Fix Notebook and Ray testcases at staging (#1274)

Fix #1271, Fix #1265, Fix #1266 ~~Not able to fix 11-similarity-search-for-motif-mining.ipynb due to #1275~~
georgia-tech-db · Oct 11, 2023 · a3b6c0c · a3b6c0c
1 parent a64d24b
commit a3b6c0c
Show file tree

Hide file tree

Showing 8 changed files with 8,777 additions and 3,315 deletions.
diff --git a/test/integration_tests/long/test_open.py b/test/integration_tests/long/test_open.py
@@ -77,4 +77,7 @@ def test_open_should_open_image(self):
                 {"testopentable.num": [1, 2], "open.data": [expected_img, expected_img]}
             )
         )
+        # We need sort. With ray enabled, the order is not guaranteed.
+        batch_res.sort_orderby(by=["testopentable.num"])
+        expected_batch.sort_orderby(by=["testopentable.num"])
         self.assertEqual(expected_batch, batch_res)
diff --git a/test/integration_tests/long/test_reuse.py b/test/integration_tests/long/test_reuse.py
@@ -91,6 +91,9 @@ def _verify_reuse_correctness(self, query, reuse_batch):
                 self.evadb, query, plan_generator=custom_plan_generator
             )
 
+        self.assertEqual(reuse_batch.columns, reuse_batch.columns)
+        reuse_batch.sort_orderby(by=[reuse_batch.columns[0]])
+        without_reuse_batch.sort_orderby(by=[reuse_batch.columns[0]])
         # printing the batches so that we can see the mismatch in the logs
         self.assertEqual(
             without_reuse_batch,

diff --git a/tutorials/01-mnist.ipynb b/tutorials/01-mnist.ipynb