def test_array_positions(self):
g = StaticPointGenerator(5)
g.prepare_positions()
g.prepare_bounds()
self.assertEqual({}, g.positions)
self.assertEqual({}, g.bounds)
def test_array_positions_with_axis(self):
g = StaticPointGenerator(5, 'repeats')
positions = [1, 2, 3, 4, 5]
bounds = [1, 2, 3, 4, 5, 6]
g.prepare_positions()
g.prepare_bounds()
self.assertEqual(positions, g.positions['repeats'].tolist())
self.assertEqual(bounds, g.bounds['repeats'].tolist())