Skip to content

Commit a80b459

Browse files
committed
tests: Update blocking tests
1 parent 5e57f1e commit a80b459

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

tests/test_dimension.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1958,21 +1958,21 @@ def test_topofusion_w_subdims_conddims(self):
19581958

19591959
# Check generated code -- expect the gsave equation to be scheduled together
19601960
# in the same loop nest with the fsave equation
1961-
bns, _ = assert_blocking(op, {'x0_blk0', 'x2_blk0', 'x1_blk0'})
1961+
bns, _ = assert_blocking(op, {'x0_blk0', 'x1_blk0', 'x2_blk0'})
19621962
exprs = FindNodes(Expression).visit(bns['x0_blk0'])
19631963
assert len(exprs) == 2
19641964
assert exprs[0].write is f
19651965
assert exprs[1].write is g
19661966

1967+
exprs = FindNodes(Expression).visit(bns['x1_blk0'])
1968+
assert len(exprs) == 1
1969+
assert exprs[0].write is h
1970+
19671971
exprs = FindNodes(Expression).visit(bns['x2_blk0'])
19681972
assert len(exprs) == 2
19691973
assert exprs[0].write is fsave
19701974
assert exprs[1].write is gsave
19711975

1972-
exprs = FindNodes(Expression).visit(bns['x1_blk0'])
1973-
assert len(exprs) == 1
1974-
assert exprs[0].write is h
1975-
19761976
def test_topofusion_w_subdims_conddims_v2(self):
19771977
"""
19781978
Like `test_topofusion_w_subdims_conddims` but with more SubDomains,

tests/test_subdomains.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -680,8 +680,8 @@ class Dummy(SubDomainSet):
680680
# Make sure it jit-compiles
681681
op.cfunction
682682

683-
assert_structure(op, ['t,n0', 't,n0,ix0_blk0,iy0_blk0,x,y,z'],
684-
't,n0,ix0_blk0,iy0_blk0,x,y,z')
683+
assert_structure(op, ['t,n0', 't,n0,x0_blk0,y0_blk0,x,y,z'],
684+
't,n0,x0_blk0,y0_blk0,x,y,z')
685685

686686
# Drag a rebuilt MultiSubDimension out of the operator
687687
dims = {d.name: d for d in FindSymbols('dimensions').visit(op)}

0 commit comments

Comments
 (0)