Frank14f/CelerisLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: construct shear remap fixture through per-level API

Browse Source
This commit is contained in:
Frank14f
2026-10-28 10:21:14 +08:00
parent 7482d95a42
commit 7b652b2423
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/validation/multigrid/test_shear_refinement_example_gpu.py
+6 -3
View File
@@ -8,14 +8,17 @@ from contextlib import ExitStack
import unittest import unittest
import numpy as np import numpy as np
from CelerisLab.cuda.context import CudaContext from CelerisLab.cuda.context import CudaContext
from research.shear_refinement.example import (CHILDREN, make, initialize, observe, from CelerisLab.lbm.refinement_stepper import FixedPatchStepper
topology, objects) from research.shear_refinement.example import (CFG, PARENTS, CHILDREN, initialize,
observe, topology, objects)
class ShearRefinementGPU(unittest.TestCase): class ShearRefinementGPU(unittest.TestCase):
def test_measured_growth_matches_explicit_manual_continuation(self): def test_measured_growth_matches_explicit_manual_continuation(self):
with CudaContext(0) as ctx, ExitStack() as owned: with CudaContext(0) as ctx, ExitStack() as owned:
old = owned.enter_context(make(ctx)) old = owned.enter_context(FixedPatchStepper(
CFG, context=ctx, refinements=[PARENTS, CHILDREN],
level_blocks=[(4, 4)] * 3, level_objects=[(), (), objects()]))
initialize(old) initialize(old)
old.step(2) old.step(2)
before = old.download() before = old.download()