def test_update_cells(self): """Test _update_cells method """ foam_mesh = FoamMesh('test_mesh', {}, self.solver, self.mesh) label = 0 cuid = foam_mesh._foamCellLabelToUuid[label] cell_f = foam_mesh.get(cuid) self.assertIsInstance(cell_f.data, DataContainer) cell = list(self.mesh.iter(item_type=CUBA.CELL))[label] self.assertEqual(cell.data, cell_f.data) updated_cells = [] for cell in foam_mesh.iter(item_type=CUBA.CELL): cell.data[CUBA.VELOCITY] = [2, 1, 3] updated_cells.append(cell) foam_mesh.update(updated_cells) label = 0 cuid = foam_mesh._foamCellLabelToUuid[label] cell_f = foam_mesh.get(cuid) self.assertIsInstance(cell_f.data, DataContainer) cell = list(self.mesh.iter(item_type=CUBA.CELL))[label] self.assertNotEqual(cell.data, cell_f.data) updated_cells = [] for cell in foam_mesh.iter(item_type=CUBA.CELL): cell.points = [ self.points[1].uid, self.points[2].uid, self.points[3].uid ] updated_cells.append(cell) with self.assertRaises(Warning): foam_mesh.update(updated_cells)
def test_update_cells(self): """Test _update_cells method """ foam_mesh = FoamMesh('test_mesh', {}, self.solver, self.mesh) label = 0 cuid = foam_mesh._foamCellLabelToUuid[label] cell_f = foam_mesh.get(cuid) self.assertIsInstance(cell_f.data, DataContainer) cell = list(self.mesh.iter(item_type=CUBA.CELL))[label] self.assertEqual(cell.data, cell_f.data) updated_cells = [] for cell in foam_mesh.iter(item_type=CUBA.CELL): cell.data[CUBA.VELOCITY] = [2, 1, 3] updated_cells.append(cell) foam_mesh.update(updated_cells) label = 0 cuid = foam_mesh._foamCellLabelToUuid[label] cell_f = foam_mesh.get(cuid) self.assertIsInstance(cell_f.data, DataContainer) cell = list(self.mesh.iter(item_type=CUBA.CELL))[label] self.assertNotEqual(cell.data, cell_f.data) updated_cells = [] for cell in foam_mesh.iter(item_type=CUBA.CELL): cell.points = [self.points[1].uid, self.points[2].uid, self.points[3].uid] updated_cells.append(cell) with self.assertRaises(Warning): foam_mesh.update(updated_cells)
def test_iter_edges(self): """Test _iter_edges method """ foam_mesh = FoamMesh('test_mesh', {}, self.solver, self.mesh) self.assertEqual(foam_mesh.iter(item_type=CUBA.EDGE), [])
def test_iter_faces(self): """Test _iter_faces method """ foam_mesh = FoamMesh('test_mesh', {}, self.solver, self.mesh) for face_f in foam_mesh.iter(item_type=CUBA.FACE): self.assertEqual(len(face_f.points), 4)
def test_iter_cells(self): """Test _iter_cells method """ foam_mesh = FoamMesh('test_mesh', {}, self.solver, self.mesh) for cell_f in foam_mesh.iter(item_type=CUBA.CELL): label = foam_mesh._uuidToFoamLabelAndType[cell_f.uid][0] cell = self.cells[label] self.assertEqual(cell.data[CUBA.VELOCITY], cell_f.data[CUBA.VELOCITY])