Skip to content

Commit 7d4cdf8

Browse files
committed
try to fix test import
1 parent 2714a97 commit 7d4cdf8

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

unit_tests/sources/declarative/partition_routers/test_cartesian_product_partition_router.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
)
2222
from airbyte_cdk.sources.types import StreamSlice
2323

24-
from .helpers import MockStream
24+
from unit_tests.sources.declarative.partition_routers.helpers import MockStream
2525

2626

2727
@pytest.mark.parametrize(

unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@
3535
from airbyte_cdk.sources.streams.checkpoint import Cursor
3636
from airbyte_cdk.sources.types import Record
3737
from airbyte_cdk.utils import AirbyteTracedException
38+
from unit_tests.sources.declarative.partition_routers.helpers import MockStream
39+
3840

3941
parent_records = [{"id": 1, "data": "data1"}, {"id": 2, "data": "data2"}]
4042
more_records = [
@@ -62,7 +64,6 @@
6264
all_parent_data_with_cursor = (
6365
data_first_parent_slice_with_cursor + data_second_parent_slice_with_cursor
6466
)
65-
from .helpers import MockStream
6667

6768

6869
class MockIncrementalStream(MockStream):

0 commit comments

Comments
 (0)