improve test coverage for data_version_diff function tests (#56)

Author: Purityj

README.md

@@ -70,18 +70,19 @@ from csvplus.load_optimized_csv import load_optimized_csv
 from csvplus.data_correction import resolve_string_value
 from csvplus.generate_report import summary_report
 
-df_v1 = load_optimized_csv("large_dataset.csv")
-df_v2 = load_optimized_csv("large_dataset2.csv")
-
-diff = data_version_diff(df_v1, df_v2)
-# Inspect the returned dictionary
-print(diff["columns_added"])
-print(diff["row_count_change"])
+# --- test data type change in csvplus.data_version_diff ---
+df1 = pd.DataFrame({"a": [1,2,3]})
+df2 = pd.DataFrame({"a": ["1","2","3"]})
+diff = data_version_diff(df1, df2)
+print(diff)
 # Optionally display a formatted summary
 display_data_version_diff(diff)
 
-resolve_string_value(df_v1, "company_name", ["Google", "Microsoft"], 80)
-summary_report(df_v1)
+# --- csvplus.data_correction --
+df_v1 = load_optimized_csv("large_dataset.csv")
+df_v2 = load_optimized_csv("large_dataset2.csv")
+resolve_string_value(df1, "company_name", ["Google", "Microsoft"], 80)
+summary_report(df1)
 ```
 
 ### Running Tests

environment.yml

@@ -6,6 +6,7 @@ dependencies:
   - rapidfuzz
   - pytest
   - numpy
+  - scipy
   - pip
   - pip:
     - faker

tests/unit/test_data_version_diff.py

@@ -130,7 +130,7 @@ def test_empty_dataframes():
     """
     Ensure the function handles empty DataFrames without error.
     """
-    
+
     df_old = pd.DataFrame()
     df_new = pd.DataFrame()
 
@@ -140,3 +140,18 @@ def test_empty_dataframes():
     assert result["row_count_change"]["new_row_count"] == 0
     assert result["columns_added"] == []
     assert result["columns_removed"] == []
+
+def test_display_data_version_diff(capsys):
+    """
+    Test that the display function runs without error and prints something.
+    """
+    df_old = pd.DataFrame({"a": [1, None]})
+    df_new = pd.DataFrame({"a": [2, None]})
+    result = data_version_diff(df_old, df_new)
+    
+    # Call the display function
+    from csvplus.data_version_diff import display_data_version_diff
+    display_data_version_diff(result)
+    
+    captured = capsys.readouterr()
+    assert "DATA VERSION CHANGE SUMMARY" in captured.out