Ejemplo n.º 1
0
        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()
Ejemplo n.º 2
0
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()