Skip to content

Commit

Permalink
chore: ran blacken
Browse files Browse the repository at this point in the history
  • Loading branch information
daniel-sanche committed Jan 19, 2024
1 parent aeea5c6 commit ef29512
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 4 deletions.
5 changes: 4 additions & 1 deletion tests/system/data/setup_fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ def admin_client():
client = Client(admin=True)
yield client


@pytest.fixture(scope="session")
def instance_id(admin_client, project_id, cluster_config):
"""
Expand Down Expand Up @@ -142,7 +143,9 @@ def table_id(
yield init_table_id
print(f"Deleting table: {parent_path}/tables/{init_table_id}")
try:
admin_client.table_admin_client.delete_table(name=f"{parent_path}/tables/{init_table_id}")
admin_client.table_admin_client.delete_table(
name=f"{parent_path}/tables/{init_table_id}"
)
except exceptions.NotFound:
print(f"Table {init_table_id} not found, skipping deletion")

Expand Down
14 changes: 11 additions & 3 deletions tests/system/data/test_system.py
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,9 @@ async def test_mutation_set_cell(table, temp_rows):
assert (await _retrieve_cell_value(table, row_key)) == new_value


@pytest.mark.skipif(bool(os.environ.get(BIGTABLE_EMULATOR)), reason="emulator doesn't use splits")
@pytest.mark.skipif(
bool(os.environ.get(BIGTABLE_EMULATOR)), reason="emulator doesn't use splits"
)
@pytest.mark.usefixtures("client")
@pytest.mark.usefixtures("table")
@retry.AsyncRetry(predicate=retry.if_exception_type(ClientError), initial=1, maximum=5)
Expand Down Expand Up @@ -811,7 +813,10 @@ async def test_read_row(table, temp_rows):
assert row.cells[0].value == b"value"


@pytest.mark.skipif(bool(os.environ.get(BIGTABLE_EMULATOR)), reason="emulator doesn't raise InvalidArgument")
@pytest.mark.skipif(
bool(os.environ.get(BIGTABLE_EMULATOR)),
reason="emulator doesn't raise InvalidArgument",
)
@pytest.mark.usefixtures("table")
@pytest.mark.asyncio
async def test_read_row_missing(table):
Expand Down Expand Up @@ -847,7 +852,10 @@ async def test_read_row_w_filter(table, temp_rows):
assert row.cells[0].labels == [expected_label]


@pytest.mark.skipif(bool(os.environ.get(BIGTABLE_EMULATOR)), reason="emulator doesn't raise InvalidArgument")
@pytest.mark.skipif(
bool(os.environ.get(BIGTABLE_EMULATOR)),
reason="emulator doesn't raise InvalidArgument",
)
@pytest.mark.usefixtures("table")
@pytest.mark.asyncio
async def test_row_exists(table, temp_rows):
Expand Down

0 comments on commit ef29512

Please sign in to comment.