def test_convert_coordinates(self):
""" Mimic click with 'anykey' pressed """
pb = PointBrowser(self.data, force_interactive=False)
pb.coordinates = [[[1, 2, 3], [4, 5, 6]]]
new_coordinates = pb._convert_coordinates()
self.assertTrue(
np.all(new_coordinates[0] == np.array([[1, 2, 3], [4, 5, 6]]).T))
def test_onclick(self):
data = np.ndarray(shape=(4, 4), dtype=float, order='F')
point = PointBrowser(data)
event = Event(xdata=0, ydata=0, key=None)
point.onclick(event)
t = point._convert_coordinates()[0]
self.assertIsInstance(t, np.ndarray)
xPoints = t[0]
self.assertIsInstance(xPoints, np.ndarray)
yPoints = t[1]
self.assertIsInstance(yPoints, np.ndarray)
self.assertEqual(xPoints[0], event.xdata, "x coordinates is set wrong")
self.assertEqual(yPoints[0], event.ydata, "y coordinates is set wrong")
def test_onclick(self):
data = np.ndarray(shape=(4, 4), dtype=float, order='F')
point = PointBrowser(data)
event = Event(xdata=0, ydata=0, key=None)
point.onclick(event)
t = point._convert_coordinates()[0]
self.assertIsInstance(t, np.ndarray)
xPoints = t[0]
self.assertIsInstance(xPoints, np.ndarray)
yPoints = t[1]
self.assertIsInstance(yPoints, np.ndarray)
self.assertEqual(xPoints[0], event.xdata, "x coordinates is set wrong")
self.assertEqual(yPoints[0], event.ydata, "y coordinates is set wrong")
def test_onclick_multilines(self):
data = np.ndarray(shape=(4, 4), dtype=float, order='F')
point = PointBrowser(data)
events = []
events.append(Event(xdata=0, ydata=0, key=None))
events.append(Event(xdata=1, ydata=0, key=None))
events.append(Event(xdata=2, ydata=2, key='AnyKeyButZorAltZ'))
events.append(Event(xdata=2, ydata=3, key=None))
for event in events:
point.onclick(event)
points = point._convert_coordinates()
self.assertEqual(len(points), 2, 'There should be two transects')
self.assertTrue(np.alltrue(points[0] == np.array([[0, 1], [0, 0]])),
't1 is not correct')
self.assertTrue(np.alltrue(points[1] == np.array([[2, 2], [2, 3]])),
't2 is not correct')
def test_onclick_multilines(self):
data = np.ndarray(shape=(4, 4), dtype=float, order='F')
point = PointBrowser(data)
events = []
events.append(Event(xdata=0, ydata=0, key=None))
events.append(Event(xdata=1, ydata=0, key=None))
events.append(Event(xdata=2, ydata=2, key='AnyKeyButZorAltZ'))
events.append(Event(xdata=2, ydata=3, key=None))
for event in events:
point.onclick(event)
points = point._convert_coordinates()
self.assertEqual(len(points), 2, 'There should be two transects')
self.assertTrue(np.alltrue(points[0] == np.array([[0, 1], [0, 0]])),
't1 is not correct')
self.assertTrue(np.alltrue(points[1] == np.array([[2, 2], [2, 3]])),
't2 is not correct')
def test_convert_coordinates(self):
""" Mimic click with 'anykey' pressed """
pb = PointBrowser(self.data, force_interactive=False)
pb.coordinates = [[[1,2,3],[4,5,6]]]
new_coordinates = pb._convert_coordinates()
self.assertTrue(np.all(new_coordinates[0] == np.array([[1,2,3], [4,5,6]]).T))
