@@ -86,19 +86,19 @@ def test_selection_mask():
86
86
assert mask [()] is True
87
87
88
88
# Explicitly deselected
89
- assert mask [( "properties" , "id" ) ] is False
89
+ assert mask ["properties" , "id" ] is False
90
90
91
91
# Missing defaults to parent selection
92
- assert mask [( "properties" , "name" ) ] is True
92
+ assert mask ["properties" , "name" ] is True
93
93
94
94
# Explicitly selected
95
- assert mask [( "properties" , "an_object" ) ] is False
95
+ assert mask ["properties" , "an_object" ] is False
96
96
97
97
# Missing defaults to parent selection
98
- assert mask [( "properties" , "an_object" , "properties" , "id" ) ] is False
98
+ assert mask ["properties" , "an_object" , "properties" , "id" ] is False
99
99
100
100
# Explicitly selected nested property
101
- assert mask [( "properties" , "an_object" , "properties" , "a_string" ) ] is True
101
+ assert mask ["properties" , "an_object" , "properties" , "a_string" ] is True
102
102
103
103
104
104
def test_metadata_mapping ():
@@ -112,27 +112,27 @@ def test_metadata_mapping():
112
112
forced_replication_method = "FULL_TABLE" ,
113
113
)
114
114
)
115
- assert mapping [( "properties" , "id" ) ] == Metadata (
115
+ assert mapping ["properties" , "id" ] == Metadata (
116
116
inclusion = Metadata .InclusionType .AUTOMATIC ,
117
117
selected = True ,
118
118
)
119
- assert mapping [( "properties" , "name" ) ] == Metadata (
119
+ assert mapping ["properties" , "name" ] == Metadata (
120
120
inclusion = Metadata .InclusionType .AVAILABLE ,
121
121
selected = True ,
122
122
)
123
- assert mapping [( "properties" , "missing" ) ] == Metadata ()
123
+ assert mapping ["properties" , "missing" ] == Metadata ()
124
124
125
125
selection_mask = mapping .resolve_selection ()
126
126
assert selection_mask [()] is True
127
- assert selection_mask [( "properties" , "id" ) ] is True
128
- assert selection_mask [( "properties" , "updated_at" ) ] is True
129
- assert selection_mask [( "properties" , "name" ) ] is True
130
- assert selection_mask [( "properties" , "missing" ) ] is True
131
- assert selection_mask [( "properties" , "an_object" ) ] is False
132
- assert selection_mask [( "properties" , "an_object" , "properties" , "nested" ) ] is False
133
- assert selection_mask [( "properties" , "not_supported_selected" ) ] is False
134
- assert selection_mask [( "properties" , "not_supported_not_selected" ) ] is False
135
- assert selection_mask [( "properties" , "selected_by_default" ) ] is True
127
+ assert selection_mask ["properties" , "id" ] is True
128
+ assert selection_mask ["properties" , "updated_at" ] is True
129
+ assert selection_mask ["properties" , "name" ] is True
130
+ assert selection_mask ["properties" , "missing" ] is True
131
+ assert selection_mask ["properties" , "an_object" ] is False
132
+ assert selection_mask ["properties" , "an_object" , "properties" , "nested" ] is False
133
+ assert selection_mask ["properties" , "not_supported_selected" ] is False
134
+ assert selection_mask ["properties" , "not_supported_not_selected" ] is False
135
+ assert selection_mask ["properties" , "selected_by_default" ] is True
136
136
137
137
138
138
def test_empty_metadata_mapping ():
@@ -264,11 +264,11 @@ def test_standard_metadata(
264
264
assert stream_metadata .schema_name == schema_name
265
265
266
266
for pk in key_properties :
267
- pk_metadata = metadata [( "properties" , pk ) ]
267
+ pk_metadata = metadata ["properties" , pk ]
268
268
assert pk_metadata .inclusion == Metadata .InclusionType .AUTOMATIC
269
269
assert pk_metadata .selected is None
270
270
271
271
for rk in valid_replication_keys or []:
272
- rk_metadata = metadata [( "properties" , rk ) ]
272
+ rk_metadata = metadata ["properties" , rk ]
273
273
assert rk_metadata .inclusion == Metadata .InclusionType .AUTOMATIC
274
274
assert rk_metadata .selected is None
0 commit comments