tests/test_audit_pandas_pickle_load.py

Rama main · réplica de lectura del repositorio original.

from scripts.audit_pandas_pickle_load import audit_source

def test_pandas_pickle_audit_ignores_csv_reads():
    assert audit_source("pd.read_csv(path)\n") == []

def test_pandas_pickle_audit_reports_pickle_reads():
    assert audit_source("pd.read_pickle(path)\n") == ["pd.read_pickle deserializes pickle data on line 1"]