def test_box(self): cells = RectCells(10, 10, 4.0, 2.0, (0.0, 0.0), 1) line = ((-2.0, -1.0), (-2.0, 0.0)) b_cells = cells.generate_cell(line) self.assertEqual(b_cells.n_y, 5) self.assertEqual(b_cells.cell_length, cells.cell_length) self.assertEqual(b_cells.cell_width, cells.cell_width) line = ((-2.0, -1.0), (0.0, -1.0)) b_cells = cells.generate_cell(line) self.assertEqual(b_cells.n_y, 1) self.assertEqual(b_cells.n_x, 5) self.assertEqual(b_cells.cell_length, cells.cell_length) self.assertEqual(b_cells.cell_width, cells.cell_width) box = [ ((-2.0, -1.0), (-2.0, 1.0)), ((-2.0, 1.0), (2.0, 1.0)), ((2.0, 1.0), (2.0, -1.0)), ((2.0, -1.0), (-2.0, -1.0)), ] for line in box: b_cells = cells.generate_cell(line) if np.abs(line[0][0] - line[1][0]) < 1.0e-6: self.assertEqual(b_cells.n_y, cells.n_y) self.assertEqual(b_cells.n_x, 1) else: self.assertEqual(b_cells.n_x, cells.n_x) self.assertEqual(b_cells.n_y, 1) self.assertEqual(b_cells.cell_length, cells.cell_length) self.assertEqual(b_cells.cell_width, cells.cell_width)