Method run

compat/src/test/python/runner_parquet.py:245–279

Run the test. Returns True if passed, False if failed.

(self)

        return 2

    def run(self) -> bool:
        """Run the test. Returns True if passed, False if failed."""
        test_name = self.__class__.__name__

        try:
            self.setup()

            result = execute_sql(self.host, self.port, f"SELECT count() FROM {self.table_name}")
            actual_count = result['dataset'][0][0]
            assert actual_count == self.expected_row_count, \
                f"Row count mismatch: expected {self.expected_row_count}, got {actual_count}"

            parquet_path = os.path.join(self.temp_dir, f"{self.table_name}.parquet")
            export_parquet(self.host, self.port, f"SELECT * FROM {self.table_name}", parquet_path)

            total_rows, num_row_groups = read_parquet(parquet_path)

            assert total_rows == self.expected_row_count, \
                f"Parquet row count mismatch: expected {self.expected_row_count}, got {total_rows}"
            if num_row_groups is not None:
                assert num_row_groups >= self.expected_min_row_groups, \
                    f"Expected at least {self.expected_min_row_groups} row groups, got {num_row_groups}"

            print(f"Test '{test_name}' passed.")
            return True

        except Exception as e:
            print(f"Test '{test_name}' failed: {e}")
            return False

        finally:
            try:
                self.teardown()
            except Exception:
                pass


class TestSymbolMultipleRowGroups(ParquetCompatTest):