def test_1d(self):
grid = Structured([-1, 2, 3, 6], (4,))
self.assertEqual(grid.get_point_count(), 4)
self.assertEqual(grid.get_cell_count(), 3)
self.assertArrayEqual(grid.get_x(), np.array([-1.0, 2.0, 3.0, 6.0]))
with self.assertRaises(IndexError):
grid.get_y()
self.assertArrayEqual(grid.get_shape(), (4,))
self.assertArrayEqual(grid.get_offset(), np.array([2, 4, 6]))
self.assertArrayEqual(grid.get_connectivity(), np.array([0, 1, 1, 2, 2, 3]))
def test_2d(self):
(x, y) = np.meshgrid([1.0, 2.0, 4.0, 8.0], [1.0, 2.0, 3.0])
grid = Structured(y.flatten(), x.flatten(), [3, 4])
self.assertEqual(grid.get_point_count(), 12)
self.assertEqual(grid.get_cell_count(), 6)
self.assertArrayEqual(
grid.get_x(),
np.array([1.0, 2.0, 4.0, 8.0, 1.0, 2.0, 4.0, 8.0, 1.0, 2.0, 4.0, 8.0]),
)
self.assertArrayEqual(
grid.get_y(),
np.array([1.0, 1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 2.0, 3.0, 3.0, 3.0, 3.0]),
)
self.assertArrayEqual(grid.get_shape(), [3, 4])
self.assertArrayEqual(grid.get_offset(), np.array([4, 8, 12, 16, 20, 24]))
self.assertArrayEqual(
grid.get_connectivity(),
np.array(
[
0,
1,
5,
4,
1,
2,
6,
5,
2,
3,
7,
6,
4,
5,
9,
8,
5,
6,
10,
9,
6,
7,
11,
10,
]
),
)