test: enhance test_register_filtered_dataframe with additional assertions for DataFrame results

Author: kosiew

python/tests/test_view.py

@@ -21,13 +21,23 @@ def test_register_filtered_dataframe():
 
     df_view = ctx.sql("SELECT * FROM view1")
 
-    results = df_view.collect()
+    filtered_results = df_view.collect()
 
-    result_dicts = [batch.to_pydict() for batch in results]
+    result_dicts = [batch.to_pydict() for batch in filtered_results]
 
     expected_results = [
         {"a": [3, 4, 5], "b": [30, 40, 50]}
     ]
 
     assert result_dicts == expected_results
 
+    df_results = df.collect()
+
+    df_result_dicts = [batch.to_pydict() for batch in df_results]
+
+    expected_df_results = [
+        {"a": [1, 2, 3, 4, 5], "b": [10, 20, 30, 40, 50]}
+    ]
+
+    assert df_result_dicts == expected_df_results
+