def test_negative_direction(self): g = LineGenerator("x", "mm", 2, -2, 5) positions = [2., 1., 0., -1., -2.] bounds = [2.5, 1.5, 0.5, -0.5, -1.5, -2.5] g.prepare_positions() g.prepare_bounds() self.assertEqual(positions, g.positions['x'].tolist()) self.assertEqual(bounds, g.bounds['x'].tolist())
def test_array_positions(self): g = LineGenerator("x", "mm", 1.0, 9.0, 5, alternate=True) positions = [1.0, 3.0, 5.0, 7.0, 9.0] bounds = [0.0, 2.0, 4.0, 6.0, 8.0, 10.0] indexes = [0, 1, 2, 3, 4] g.prepare_positions() g.prepare_bounds() self.assertEqual(positions, g.positions['x'].tolist()) self.assertEqual(bounds, g.bounds['x'].tolist())
def test_array_positions_single_point(self): g = LineGenerator(["x", "y"], "mm", [1.0, -3.0], [5.0, -4.0], 1) g.prepare_positions() g.prepare_bounds() x_positions = [3.0] y_positions = [-3.5] x_bounds = [1.0, 5.0] y_bounds = [-3.0, -4.0] self.assertEqual(x_positions, g.positions['x'].tolist()) self.assertEqual(y_positions, g.positions['y'].tolist()) self.assertEqual(x_bounds, g.bounds['x'].tolist()) self.assertEqual(y_bounds, g.bounds['y'].tolist())
def test_array_positions(self): g = LineGenerator(["x", "y"], "mm", [1.0, 2.0], [5.0, 10.0], 5) g.prepare_positions() g.prepare_bounds() x_positions = [1.0, 2.0, 3.0, 4.0, 5.0] y_positions = [2.0, 4.0, 6.0, 8.0, 10.0] x_bounds = [0.5, 1.5, 2.5, 3.5, 4.5, 5.5] y_bounds = [1, 3, 5, 7, 9, 11] self.assertEqual(x_positions, g.positions['x'].tolist()) self.assertEqual(y_positions, g.positions['y'].tolist()) self.assertEqual(x_bounds, g.bounds['x'].tolist()) self.assertEqual(y_bounds, g.bounds['y'].tolist())
def test_single_point_exact(self): g = LineGenerator("x", "mm", 1.0, 1.0, 1) g.prepare_positions() g.prepare_bounds() self.assertEqual([1.0], g.positions["x"].tolist()) self.assertEqual([1.0, 1.0], g.bounds["x"].tolist())