Skip to content

Commit

Permalink
lint fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
matthiasdiener authored Feb 5, 2024
1 parent 544b677 commit 3fd38a1
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions test/test_pytato.py
Original file line number Diff line number Diff line change
Expand Up @@ -1117,7 +1117,7 @@ def test_dot_visualizers():

# {{{ Test PytatoKeyBuilder

def run_test_with_new_python_invocation(f, *args, extra_env_vars = None) -> None:
def run_test_with_new_python_invocation(f, *args, extra_env_vars=None) -> None:
import os
if extra_env_vars is None:
extra_env_vars = {}
Expand All @@ -1140,6 +1140,8 @@ def run_test_with_new_python_invocation(f, *args, extra_env_vars = None) -> None
def run_test_with_new_python_invocation_inner() -> None:
from base64 import b64decode
from pickle import loads
import os

f, args = loads(b64decode(os.environ["INVOCATION_INFO"].encode()))

f(*args)
Expand Down Expand Up @@ -1174,11 +1176,12 @@ def test_persistent_hashing_and_persistent_dict() -> None:
pd[dag] = 42

# Make sure that the key stays the same across Python invocations
run_test_with_new_python_invocation(_test_persistent_hashing_and_persistent_dict_stage2,
tmpdir)
run_test_with_new_python_invocation(
_test_persistent_hashing_and_persistent_dict_stage2, tmpdir)
finally:
shutil.rmtree(tmpdir)


def _test_persistent_hashing_and_persistent_dict_stage2(tmpdir) -> None:
from pytools.persistent_dict import WriteOncePersistentDict, ReadOnlyEntryError

Expand Down

0 comments on commit 3fd38a1

Please sign in to comment.