diff --git a/tests/activities/test_gates.py b/tests/activities/test_gates.py index 366829a..ab5b0a4 100644 --- a/tests/activities/test_gates.py +++ b/tests/activities/test_gates.py @@ -32,7 +32,9 @@ async def test_data_quality_gate_with_null_values_filter_discard(gates_fixture): # Setup test data input_data = { 'filters': { - 'NULL_VALUES_FILTER': 'DISCARD' + 'NULL_VALUES_FILTER': { + 'policy': 'DISCARD' + } }, 'data': { 'tag': ['tag1', 'tag2', 'tag3'], @@ -62,7 +64,9 @@ async def test_data_quality_gate_with_out_of_bounds_filter_keep(gates_fixture): # Setup test data with out of bounds values input_data = { 'filters': { - 'OUT_OF_BOUNDS_FILTER': 'KEEP' + 'OUT_OF_BOUNDS_FILTER': { + 'policy': 'KEEP' + } }, 'data': { 'tag': ['tag1', 'tag2', 'tag3'], @@ -95,8 +99,12 @@ async def test_data_quality_gate_with_multiple_filters(gates_fixture): # Setup test data input_data = { 'filters': { - 'NULL_VALUES_FILTER': 'DISCARD', - 'OUT_OF_BOUNDS_FILTER': 'DISCARD' + 'NULL_VALUES_FILTER': { + 'policy': 'DISCARD' + }, + 'OUT_OF_BOUNDS_FILTER': { + 'policy': 'DISCARD' + } }, 'data': { 'tag': ['tag1', 'tag2', 'tag3', 'tag4'], @@ -129,7 +137,9 @@ async def test_data_quality_gate_with_unknown_filter(gates_fixture): gates_fixture.warning = MagicMock() input_data = { 'filters': { - 'UNKNOWN_FILTER': 'DISCARD' + 'UNKNOWN_FILTER': { + 'policy': 'DISCARD' + } }, 'data': { 'tag': ['tag1'], @@ -160,7 +170,9 @@ async def test_data_quality_gate_with_filter_error(gates_fixture): # Setup test data input_data = { 'filters': { - 'NULL_VALUES_FILTER': 'DISCARD' + 'NULL_VALUES_FILTER': { + 'policy': 'DISCARD' + } }, 'data': { 'tag': ['tag1'], @@ -199,7 +211,9 @@ async def test_data_quality_gate_with_empty_data(gates_fixture): # Setup empty input data input_data = { 'filters': { - 'NULL_VALUES_FILTER': 'DISCARD' + 'NULL_VALUES_FILTER': { + 'policy': 'DISCARD' + } }, 'data': { 'tag': [],