Skip to content

Commit d1cc801

Browse files
committed
integration tests
1 parent b996515 commit d1cc801

1 file changed

Lines changed: 19 additions & 1 deletion

File tree

‎tests/integration/cag/test_inequality.py‎

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1059,12 +1059,30 @@ def test_datetime_values_are_clipped_to_min_max_in_constraint():
10591059
# Run
10601060
synthesizer = run_copula(data, metadata, constraints=[constraint])
10611061
synthetic_data = synthesizer.sample('fake_hotel_guests', len(data))
1062-
diagnostic_report = run_diagnostic(data, synthetic_data, metadata)
1062+
diagnostic_report = run_diagnostic(data, synthetic_data, metadata, [constraint.to_dict()])
10631063

10641064
# Assert
1065+
expected_properties = pd.DataFrame({
1066+
'Property': [
1067+
'Data Validity',
1068+
'Data Structure',
1069+
'Constraint Validity',
1070+
],
1071+
'Score': [1.0, 1.0, 1.0],
1072+
})
10651073
metadata.validate_data(synthetic_data)
10661074
synthesizer.validate(synthetic_data)
10671075
assert diagnostic_report.get_score() == 1.0
1076+
pd.testing.assert_frame_equal(diagnostic_report.get_properties(), expected_properties)
1077+
assert constraint.to_dict() == {
1078+
'class_name': 'Inequality',
1079+
'parameters': {
1080+
'low_column_name': 'checkin_date',
1081+
'high_column_name': 'checkout_date',
1082+
'table_name': 'fake_hotel_guests',
1083+
'strict_boundaries': False,
1084+
},
1085+
}
10681086
data = data['fake_hotel_guests']
10691087
synthetic_data = synthetic_data['fake_hotel_guests']
10701088
for col in ['checkin_date', 'checkout_date']:

0 commit comments

Comments
 (0)