Set pandas output width

chdb-io · Jun 19, 2024 · 7a0c3dc · 7a0c3dc
1 parent 3860a95
commit 7a0c3dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_on_df.py b/tests/test_on_df.py
@@ -1,10 +1,10 @@
 import atexit
 import io
 import os.path
-import sys
 import time
 import unittest
 
+import pandas as pd
 from chdb.dataframe import Table, pandas_read_parquet
 from utils import current_dir
 
@@ -40,6 +40,8 @@
 # run print at exit
 atexit.register(lambda: print("\n" + output.getvalue()))
 
+pd.set_option("display.max_columns", 10)
+pd.set_option("display.width", 200)
 
 class TestRunOnDf(unittest.TestCase):