diff --git a/python/pyspark/sql/tests/connect/test_connect_basic.py b/python/pyspark/sql/tests/connect/test_connect_basic.py index 17d50f0f50e9b..11d51f734766b 100644 --- a/python/pyspark/sql/tests/connect/test_connect_basic.py +++ b/python/pyspark/sql/tests/connect/test_connect_basic.py @@ -128,6 +128,12 @@ def test_limit_offset(self): pd2 = df.offset(98).limit(10).toPandas() self.assertEqual(2, len(pd2.index)) + def test_head(self): + df = self.connect.read.table(self.tbl_name) + pd = df.head(10) + self.assertIsNotNone(pd) + self.assertEqual(10, len(pd.index)) + def test_simple_datasource_read(self) -> None: writeDf = self.df_text tmpPath = tempfile.mkdtemp()