|
5 | 5 | from unittest.mock import AsyncMock, MagicMock, call, patch |
6 | 6 |
|
7 | 7 | import pytest |
8 | | -from sqlmodel import ( # Import Relationship if needed for model definition |
9 | | - select, |
10 | | -) |
| 8 | +from sqlmodel import \ |
| 9 | + select # Import Relationship if needed for model definition |
11 | 10 | from sqlmodel.ext.asyncio.session import AsyncSession |
12 | 11 |
|
13 | | -from sqlmodel_crud_utils.a_sync import ( |
14 | | - bulk_upsert_mappings as async_bulk_upsert_mappings, |
15 | | -) |
| 12 | +from sqlmodel_crud_utils.a_sync import \ |
| 13 | + bulk_upsert_mappings as async_bulk_upsert_mappings |
16 | 14 | from sqlmodel_crud_utils.a_sync import delete_row as async_delete_row |
17 | | -from sqlmodel_crud_utils.a_sync import ( |
18 | | - get_one_or_create as async_get_one_or_create, |
19 | | -) |
20 | | -from sqlmodel_crud_utils.a_sync import ( |
21 | | - get_result_from_query as async_get_result_from_query, |
22 | | -) |
| 15 | +from sqlmodel_crud_utils.a_sync import \ |
| 16 | + get_one_or_create as async_get_one_or_create |
| 17 | +from sqlmodel_crud_utils.a_sync import \ |
| 18 | + get_result_from_query as async_get_result_from_query |
23 | 19 | from sqlmodel_crud_utils.a_sync import get_row as async_get_row |
24 | 20 | from sqlmodel_crud_utils.a_sync import get_rows as async_get_rows |
25 | | -from sqlmodel_crud_utils.a_sync import ( |
26 | | - get_rows_within_id_list as async_get_rows_within_id_list, |
27 | | -) |
28 | | -from sqlmodel_crud_utils.a_sync import ( |
29 | | - insert_data_rows as async_insert_data_rows, |
30 | | -) |
| 21 | +from sqlmodel_crud_utils.a_sync import \ |
| 22 | + get_rows_within_id_list as async_get_rows_within_id_list |
| 23 | +from sqlmodel_crud_utils.a_sync import \ |
| 24 | + insert_data_rows as async_insert_data_rows |
31 | 25 | from sqlmodel_crud_utils.a_sync import update_row as async_update_row |
32 | 26 | from sqlmodel_crud_utils.a_sync import write_row as async_write_row |
33 | 27 |
|
34 | 28 | from .conftest import MockModelFactory, MockRelatedModelFactory |
35 | | -from .models import ( # Assuming MockRelatedModel is needed for relationship |
36 | | - MockModel, |
37 | | -) |
| 29 | +from .models import \ |
| 30 | + MockModel # Assuming MockRelatedModel is needed for relationship |
38 | 31 |
|
39 | 32 | # --- Tests for get_result_from_query --- |
40 | 33 | # (No changes needed for these tests - they use real sessions) |
|
0 commit comments