Fix data link

chdb-io · Mar 20, 2024 · 5eeef64 · 5eeef64
1 parent 9d2ed89
commit 5eeef64
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_issue31.py b/tests/test_issue31.py
@@ -10,7 +10,7 @@
 
 from timeout_decorator import timeout
 
-csv_url = "https://media.githubusercontent.com/media/datablist/sample-csv-files/main/files/organizations/organizations-2000000.zip"
+csv_url = "https://github.com/chdb-io/chdb/files/14662379/organizations-500000.zip"
 
 
 # download csv file, and unzip it
@@ -33,7 +33,7 @@ def download_and_extract(url, save_path):
 def payload():
     now = time.time()
     res = chdb.query(
-        'select Name, count(*) cnt from file("organizations-2000000.csv", CSVWithNames) group by Name order by cnt desc, Name asc limit 10000',
+        'select Name, count(*) cnt from file("organizations-500000.csv", CSVWithNames) group by Name order by cnt desc, Name asc limit 10000',
         "CSV",
     )
     # calculate md5 of the result
@@ -71,11 +71,11 @@ def handler(signum, frame):
 
 class TestAggOnCSVSpeed(unittest.TestCase):
     def setUp(self):
-        download_and_extract(csv_url, "organizations-2000000.zip")
+        download_and_extract(csv_url, "organizations-500000.zip")
 
     def tearDown(self):
-        os.remove("organizations-2000000.csv")
-        os.remove("organizations-2000000.zip")
+        os.remove("organizations-500000.csv")
+        os.remove("organizations-500000.zip")
 
     def _test_agg(self, arg=None):
         payload()