def test_add_lines(self): fig, ax = pl.subplots() x = np.arange(0, 10) y = np.arange(0, 10) xy = np.dstack((x, y)) vis.add_lines(ax, xy) vis.add_lines(ax, np.array([xy]))