def test_iter_edges(self): """Test iter_edges method """ num_mesh = NumerrinMesh('test_mesh', self.mesh, self.pool) self.assertEqual(sum(1 for _ in num_mesh.iter(item_type=CUBA.EDGE)), num_mesh.count_of(CUBA.EDGE))
def test_count_of(self): """Test count_of method """ num_mesh = NumerrinMesh('test_mesh', self.mesh, self.pool) item_type = CUBA.POINT self.assertEqual(num_mesh.count_of(item_type), self.mesh.count_of(item_type)) item_type = CUBA.EDGE self.assertEqual(num_mesh.count_of(item_type), 12) item_type = CUBA.FACE self.assertEqual(num_mesh.count_of(item_type), self.mesh.count_of(item_type)) item_type = CUBA.CELL self.assertEqual(num_mesh.count_of(item_type), self.mesh.count_of(item_type))