def test_adding_to_shape_list(): """Test adding shapes to ShapeList.""" np.random.seed(0) data = 20 * np.random.random((4, 2)) shape = Rectangle(data) shape_list = ShapeList() shape_list.add(shape) assert len(shape_list.shapes) == 1 assert shape_list.shapes[0] == shape
def test_bad_color_array(attribute): """Test adding shapes to ShapeList.""" np.random.seed(0) data = 20 * np.random.random((4, 2)) shape = Rectangle(data) shape_list = ShapeList() shape_list.add(shape) # test setting color with a color array of the wrong shape bad_color_array = np.array([[0, 0, 0, 1], [1, 1, 1, 1]]) with pytest.raises(ValueError): setattr(shape_list, f'{attribute}_color', bad_color_array)
def test_nD_shapes(): """Test adding shapes to ShapeList.""" np.random.seed(0) shape_list = ShapeList() data = 20 * np.random.random((6, 3)) data[:, 0] = 0 shape_a = Polygon(data) shape_list.add(shape_a) data = 20 * np.random.random((6, 3)) data[:, 0] = 1 shape_b = Path(data) shape_list.add(shape_b) assert len(shape_list.shapes) == 2 assert shape_list.shapes[0] == shape_a assert shape_list.shapes[1] == shape_b assert shape_list._vertices.shape[1] == 2 assert shape_list._mesh.vertices.shape[1] == 2 shape_list.ndisplay = 3 assert shape_list._vertices.shape[1] == 3 assert shape_list._mesh.vertices.shape[1] == 3
def test_empty_shape_list(): """Test instantiating empty ShapeList.""" shape_list = ShapeList() assert len(shape_list.shapes) == 0