Fix Notebook and Ray testcases at staging (georgia-tech-db#1274)

Fix georgia-tech-db#1271, Fix georgia-tech-db#1265, Fix georgia-tech-db#1266 ~~Not able to fix 11-similarity-search-for-motif-mining.ipynb due to georgia-tech-db#1275~~
RichardZhangRZ · Oct 23, 2023 · 4a83940 · 4a83940
1 parent 98a8466
commit 4a83940
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