File tree Expand file tree Collapse file tree 2 files changed +9
-3
lines changed
airbyte_cdk/test/standard_tests Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -354,6 +354,11 @@ def test_docker_image_build_and_read(
354354 set (streams_list ) - set (stream .name for stream in scenario .empty_streams )
355355 )
356356
357+ if scenario .empty_streams :
358+ # Filter out streams marked as empty in the scenario.
359+ empty_stream_names = [stream .name for stream in scenario .empty_streams ]
360+ streams_list = [s for s in streams_list if s .name not in empty_stream_names ]
361+
357362 configured_catalog : ConfiguredAirbyteCatalog = ConfiguredAirbyteCatalog (
358363 streams = [
359364 ConfiguredAirbyteStream (
Original file line number Diff line number Diff line change @@ -123,9 +123,10 @@ def test_basic_read(
123123 streams = discover_result .catalog .catalog .streams # type: ignore [reportOptionalMemberAccess, union-attr]
124124
125125 if scenario .empty_streams :
126- # Don't read from any streams marked as empty in the scenario.
127- streams = list (set (streams ) - set (stream .name for stream in scenario .empty_streams ))
128-
126+ # Filter out streams marked as empty in the scenario.
127+ empty_stream_names = [stream .name for stream in scenario .empty_streams ]
128+ streams = [s for s in streams if s .name not in empty_stream_names ]
129+
129130 configured_catalog = ConfiguredAirbyteCatalog (
130131 streams = [
131132 ConfiguredAirbyteStream (
You can’t perform that action at this time.
0 commit comments