Skip to content

Commit

Permalink
Try to address ResourceWarnings
Browse files Browse the repository at this point in the history
  • Loading branch information
mroeschke committed May 3, 2024
1 parent de07956 commit 1a8b8ad
Showing 1 changed file with 14 additions and 10 deletions.
24 changes: 14 additions & 10 deletions python/custreamz/custreamz/tests/test_dataframes.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,19 +24,23 @@

@pytest.fixture(scope="module")
def client():
client = Client(processes=False, asynchronous=False)
try:
with Client(processes=False, asynchronous=False) as client:
yield client
finally:
client.close()


@pytest.fixture(params=["core", "dask"])
def stream(request, client):
if request.param == "core":
return Stream()
else:
return DaskStream()
@pytest.fixture(scope="module")
def dask_stream(client):
return DaskStream()


@pytest.fixture
def core_stream():
return Stream()


@pytest.fixture(params=["core_stream", "dask_stream"])
def stream(request):
return request.getfixturevalue(request.param)


def test_identity(stream):
Expand Down

0 comments on commit 1a8b8ad

Please sign in to comment.