def test_connect_face_rows(self):
"""in order"""
geometry_array = np.array([
[1, 1, 2],
[2, 2, 3],
[3, 3, 4],
[4, 4, 5],
[5, 5, 6],
])
nedges = geometry_array.shape[0]
results_array = np.arange(0, nedges)
#print(results_array)
iedges, unused_geometry_arrays2, unused_results_arrays2 = connect_face_rows(
geometry_array, results_array, skip_cleanup=False)
assert np.array_equal(iedges, [[0, 1, 2, 3, 4]]), 'iedges=%s' % iedges
#-----------------------------------------------------------------------
# out of order
geometry_array = np.array([
[1, 1, 2], # 0
[2, 4, 5], # 3
[3, 5, 6], # 4
[4, 3, 4], # 2
[5, 2, 3], # 1
])
nedges = geometry_array.shape[0]
results_array = np.arange(0, nedges)
iedges, unused_geometry_arrays2, unused_results_arrays2 = connect_face_rows(
geometry_array, results_array, skip_cleanup=False)
assert np.array_equal(iedges, [[0, 4, 3, 1, 2]]), 'iedges=%s' % iedges
#print(geometry_array2)
#-----------------------------------------------------------------------
# in order, two blocks
#print('*****************')
geometry_array = np.array([
# block 1
[1, 1, 2],
[2, 2, 3],
[3, 3, 4],
# block 2
[10, 10, 20],
[20, 20, 30],
[30, 30, 40],
])
nedges = geometry_array.shape[0]
results_array = np.arange(0, nedges)
#print(results_array)
iedges, unused_geometry_array2, unused_results_array2 = connect_face_rows(
geometry_array, results_array, skip_cleanup=False)
assert np.array_equal(iedges,
[[0, 1, 2], [3, 4, 5]]), 'iedges=%s' % iedges
def test_connect_face_rows_ring_1(self):
"""in order, one ring"""
geometry_array = np.array([
[1, 1, 2],
[2, 2, 3],
[3, 3, 4],
[4, 1, 4],
])
nedges = geometry_array.shape[0]
results_array = np.arange(0, nedges)
#print(results_array)
iedges, unused_geometry_array2, unused_results_array2 = connect_face_rows(
geometry_array, results_array, skip_cleanup=False)
assert np.array_equal(iedges, [[0, 1, 2, 3, 0]]), 'iedges=%s' % iedges
def test_connect_face_rows_ring_2(self):
"""in order, two rings"""
geometry_array = np.array([
[1, 1, 2],
[2, 2, 3],
[3, 3, 4],
[4, 1, 4],
[10, 10, 20],
[20, 20, 30],
[30, 30, 40],
[40, 10, 40],
])
nedges = geometry_array.shape[0]
results_array = np.arange(0, nedges)
#print(results_array)
iedges, geometry_array2, results_array2 = connect_face_rows(
geometry_array, results_array, skip_cleanup=False)
assert np.array_equal(
iedges, [[0, 1, 2, 3, 0], [4, 5, 6, 7, 4]]), 'iedges=%s' % iedges
