def test_face_centered_grid2_for_each():
a = pyjet.FaceCenteredGrid2((10, 10))
# Workaround for Python 2.x which doesn't support nonlocal
d = {'ei': 0, 'ej': 0}
def checkU(i, j):
i == d['ei']
j == d['ej']
d['ei'] += 1
if d['ei'] >= 11:
d['ei'] = 0
d['ej'] += 1
a.forEachUIndex(checkU)
d = {'ei': 0, 'ej': 0}
def checkV(i, j):
i == d['ei']
j == d['ej']
d['ei'] += 1
if d['ei'] >= 10:
d['ei'] = 0
d['ej'] += 1
a.forEachVIndex(checkV)
def testForEach(self):
a = pyjet.FaceCenteredGrid2((10, 10))
# Workaround for Python 2.x which doesn't support nonlocal
d = {'ei': 0, 'ej': 0}
def checkU(i, j):
self.assertEqual(i, d['ei'])
self.assertEqual(j, d['ej'])
d['ei'] += 1
if d['ei'] >= 11:
d['ei'] = 0
d['ej'] += 1
a.forEachUIndex(checkU)
d = {'ei': 0, 'ej': 0}
def checkV(i, j):
self.assertEqual(i, d['ei'])
self.assertEqual(j, d['ej'])
d['ei'] += 1
if d['ei'] >= 10:
d['ei'] = 0
d['ej'] += 1
a.forEachVIndex(checkV)
def test_face_centered_grid2_serialization():
a = pyjet.FaceCenteredGrid2((10, 10))
def filler(pt):
return (pt.x, pt.y)
a.fill(filler)
flatBuffer = a.serialize()
b = pyjet.FaceCenteredGrid2()
b.deserialize(flatBuffer)
for j in range(10):
for i in range(11):
b.u(i, j) == i
for j in range(11):
for i in range(10):
b.v(i, j) == i
def testSerialization(self):
a = pyjet.FaceCenteredGrid2((10, 10))
def filler(pt):
return (pt.x, pt.y)
a.fill(filler)
flatBuffer = a.serialize()
b = pyjet.FaceCenteredGrid2()
b.deserialize(flatBuffer)
for j in range(10):
for i in range(11):
self.assertEqual(b.u(i, j), i)
for j in range(11):
for i in range(10):
self.assertEqual(b.v(i, j), j)
def test_face_centered_grid2_fill():
a = pyjet.FaceCenteredGrid2((10, 10))
a.fill((3.0, 4.0))
for j in range(10):
for i in range(11):
assert a.u(i, j) == 3.0
for j in range(11):
for i in range(10):
assert a.v(i, j) == 4.0
def filler(pt):
return (pt.x, pt.y)
a.fill(filler)
for j in range(10):
for i in range(11):
a.u(i, j) == i
for j in range(11):
for i in range(10):
a.v(i, j) == j
def testFill(self):
a = pyjet.FaceCenteredGrid2((10, 10))
a.fill((3.0, 4.0))
for j in range(10):
for i in range(11):
self.assertEqual(a.u(i, j), 3.0)
for j in range(11):
for i in range(10):
self.assertEqual(a.v(i, j), 4.0)
def filler(pt):
return (pt.x, pt.y)
a.fill(filler)
for j in range(10):
for i in range(11):
self.assertEqual(a.u(i, j), i)
for j in range(11):
for i in range(10):
self.assertEqual(a.v(i, j), j)