diff --git a/posthog/hogql/database/test/test_database.py b/posthog/hogql/database/test/test_database.py index edd55d27578df..0b5f6b6cafd46 100644 --- a/posthog/hogql/database/test/test_database.py +++ b/posthog/hogql/database/test/test_database.py @@ -33,12 +33,14 @@ def test_can_select_from_each_table_at_all(self, poe_enabled: bool) -> None: serialized_database = serialize_database(create_hogql_database(team_id=self.team.pk)) for table, possible_columns in serialized_database.items(): if table == "numbers": - continue - - columns = [ - x["key"] for x in possible_columns if "table" not in x and "chain" not in x and "fields" not in x - ] - execute_hogql_query(f"SELECT {','.join(columns)} FROM {table}", team=self.team) + execute_hogql_query("SELECT number FROM numbers(10) LIMIT 100", self.team) + else: + columns = [ + x["key"] + for x in possible_columns + if "table" not in x and "chain" not in x and "fields" not in x + ] + execute_hogql_query(f"SELECT {','.join(columns)} FROM {table}", team=self.team) @patch("posthog.hogql.query.sync_execute", return_value=(None, None)) @pytest.mark.usefixtures("unittest_snapshot")