From 54a007d6a13e54be8fe48297a117c0541f6b97a8 Mon Sep 17 00:00:00 2001 From: Shabnam Khan Date: Wed, 28 Jun 2023 13:47:28 +0530 Subject: [PATCH] Modified the shard_count --- src/test/regress/citus_tests/test/test_auto_shard_split.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/test/regress/citus_tests/test/test_auto_shard_split.py b/src/test/regress/citus_tests/test/test_auto_shard_split.py index 14d20bab4..40fa4a2d3 100644 --- a/src/test/regress/citus_tests/test/test_auto_shard_split.py +++ b/src/test/regress/citus_tests/test/test_auto_shard_split.py @@ -6,7 +6,7 @@ def test_isolate(cluster): cluster.coordinator.restart() cluster.coordinator.sql("SET citus.split_shard_group_size_threshold to 100") cluster.coordinator.sql("CREATE TABLE test_row(i integer)") - cluster.coordinator.sql("SELECT create_distributed_table('test_row','i')") + cluster.coordinator.sql("SELECT create_distributed_table('test_row','i',shard_count:=1)") cluster.coordinator.sql("INSERT INTO test_row SELECT 1 from generate_series(1,20000) ") row_count1 = cluster.coordinator.sql_value("SELECT count(*) FROM pg_dist_shard") @@ -25,9 +25,8 @@ def test_split(cluster): cluster.coordinator.restart() cluster.coordinator.sql("SET citus.split_shard_group_size_threshold to 100") - cluster.coordinator.sql("SET citus.shard_count to 1") cluster.coordinator.sql("CREATE TABLE test_row(i integer)") - cluster.coordinator.sql("SELECT create_distributed_table('test_row','i')") + cluster.coordinator.sql("SELECT create_distributed_table('test_row','i',shard_count:=1)") cluster.coordinator.sql("INSERT INTO test_row SELECT generate_series(1,20000) ") row_count1 = cluster.coordinator.sql_value("SELECT count(*) FROM pg_dist_shard") @@ -39,4 +38,4 @@ def test_split(cluster): row_count2 = cluster.coordinator.sql_value("SELECT count(*) FROM pg_dist_shard") - assert row_count2>row_count1, "Shard didn't get split" \ No newline at end of file + assert row_count2==row_count1+1, "Shard didn't get split" \ No newline at end of file