SIENTIAPDE-1110
Refactor test cases in test_gates.py to update filter configurations, extracting policy into a dictionary format for improved clarity and consistency in data quality gate tests.
This commit is contained in:
@@ -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': [],
|
||||
|
||||
Reference in New Issue
Block a user