blocks = 1 ctrl = LBSimulationController(SimulationTest, SphereGeometry) ctrl.run(ignore_cmdline=True) cls.digits = io.filename_iter_digits(MAX_ITERS) cls.ref = np.load(io.filename(output, cls.digits, 0, MAX_ITERS)) def test_horiz_2blocks(self): global blocks, output output = os.path.join(tmpdir, 'horiz_2block') blocks = 2 ctrl = LBSimulationController(SimulationTest, SphereGeometry) ctrl.run(ignore_cmdline=True) util.verify_fields(self.ref, output, self.digits, MAX_ITERS) def setUpModule(): global tmpdir tmpdir = tempfile.mkdtemp() def tearDownModule(): shutil.rmtree(tmpdir) if __name__ == '__main__': args = util.parse_cmd_line() block_size = args.block_size if block_size < mem_align: mem_align = block_size unittest.main()
class TestInterblockPropagation(unittest.TestCase): @classmethod def setUpClass(cls): global blocks, output output = os.path.join(tmpdir, 'ref') blocks = 1 LBSimulationController(SimulationTest, LDCGeometry).run(ignore_cmdline=True) cls.digits = io.filename_iter_digits(MAX_ITERS) cls.ref = np.load(io.filename(output, cls.digits, 0, MAX_ITERS)) def test_8blocks(self): global blocks, output output = os.path.join(tmpdir, 'horiz_8block') blocks = 8 ctrl = LBSimulationController(SimulationTest, LDCGeometry) ctrl.run(ignore_cmdline=True) util.verify_fields(self.ref, output, self.digits, MAX_ITERS) def tearDownModule(): shutil.rmtree(tmpdir) if __name__ == '__main__': args = util.parse_cmd_line() block_size = args.block_size access_pattern = args.access_pattern if block_size < mem_align: mem_align = block_size unittest.main()