Skip to content

Commit b434364

Browse files
author
superpios
committed
fix(tests): colonne DataFrame allineate a 5 campi per REGOLA-001
1 parent 28a38dd commit b434364

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/test_engine.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def test_regola_001_soglia():
3030
["MARIO ROSSI", "E4", "2025", "inc", "a4"],
3131
["MARIO ROSSI", "E5", "2025", "inc", "a5"],
3232
]
33-
df = _df(rows, ["person_name", "entity_id", "year"])
33+
df = _df(rows, ["person_name", "entity_id", "year", "source_dataset", "source_record_id"])
3434
leads = apply_regola_001(df, 5, "same_calendar_year")
3535
assert len(leads) == 1
3636
assert leads[0]["rule_id"] == "REGOLA-001"
@@ -45,7 +45,7 @@ def test_regola_001_sotto_soglia():
4545
["MARIO ROSSI", "E3", "2025", "inc", "a3"],
4646
["MARIO ROSSI", "E4", "2025", "inc", "a4"],
4747
]
48-
df = _df(rows, ["person_name", "entity_id", "year"])
48+
df = _df(rows, ["person_name", "entity_id", "year", "source_dataset", "source_record_id"])
4949
assert apply_regola_001(df, 5, "same_calendar_year") == []
5050

5151

@@ -93,7 +93,7 @@ def test_determinismo_id_e_data():
9393
["MARIO ROSSI", "E4", "2025", "inc", "a4"],
9494
["MARIO ROSSI", "E5", "2025", "inc", "a5"],
9595
]
96-
df = _df(rows, ["person_name", "entity_id", "year"])
96+
df = _df(rows, ["person_name", "entity_id", "year", "source_dataset", "source_record_id"])
9797
a = apply_regola_001(df, 5, "same_calendar_year")
9898
b = apply_regola_001(df, 5, "same_calendar_year")
9999
assert json.dumps(a, sort_keys=True) == json.dumps(b, sort_keys=True)

0 commit comments

Comments
 (0)