Skip to content

Commit

Permalink
fixed sympy tests
Browse files Browse the repository at this point in the history
  • Loading branch information
cod3monk committed Feb 17, 2022
1 parent af801f2 commit 2438d74
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_layer_condition.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
from sympy import oo

from kerncraft import kerncraft as kc
from kerncraft.kernel import symbol_pos_int
from kerncraft.prefixedunit import PrefixedUnit


Expand Down Expand Up @@ -66,7 +67,7 @@ def test_3d_7pt(self):
with open(store_file, 'rb') as f:
results = pickle.load(f)
result = next(iter(results.values()))
N, M, i, j, k = sympy.var('N, M, i, j, k')
N, M, i, j, k = [symbol_pos_int(c) for c in 'NMijk']
result_expected = {'accesses':
{'a': [(k - 1, j, i),
(k, j - 1, i),
Expand Down Expand Up @@ -167,7 +168,7 @@ def test_constantdim(self):
results = pickle.load(f)
result = next(iter(results.values()))

N, M, j, i = sympy.var('N'), sympy.var('M'), sympy.var('j'), sympy.var('i')
N, M, j, i = [symbol_pos_int(c) for c in 'NMji']
result_expected = \
{'accesses': {'W': [(0, j, i), (1, j, i)],
'a': [(j - 1, i), (j, i - 1), (j, i), (j, i + 1), (j + 1, i)],
Expand Down

0 comments on commit 2438d74

Please sign in to comment.