From 2ab52808d39eba737a592b5385cd9dcb1871b407 Mon Sep 17 00:00:00 2001
From: Luke Campagnola <luke.campagnola@gmail.com>
Date: Tue, 3 May 2016 09:20:23 -0700
Subject: [PATCH] added simple roi tests (these do not check output)

---
 pyqtgraph/graphicsItems/tests/test_ROI.py | 54 +++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 pyqtgraph/graphicsItems/tests/test_ROI.py

diff --git a/pyqtgraph/graphicsItems/tests/test_ROI.py b/pyqtgraph/graphicsItems/tests/test_ROI.py
new file mode 100644
index 00000000..15901490
--- /dev/null
+++ b/pyqtgraph/graphicsItems/tests/test_ROI.py
@@ -0,0 +1,54 @@
+import pyqtgraph as pg
+pg.mkQApp()
+
+vb = pg.ViewBox()
+data = pg.np.ones((7, 100, 110, 5))
+image_tx = pg.ImageItem(data[:, :, 0, 0])
+image_xy = pg.ImageItem(data[0, :, :, 0])
+image_yz = pg.ImageItem(data[0, 0, :, :])
+vb.addItem(image_tx)
+vb.addItem(image_xy)
+vb.addItem(image_yz)
+
+size = (10, 15)
+pos = (0, 0)
+rois = [
+    pg.ROI(pos, size),
+    pg.RectROI(pos, size),
+    pg.EllipseROI(pos, size),
+    pg.CircleROI(pos, size),
+    pg.PolyLineROI([pos, size]),
+]
+
+for roi in rois:
+    vb.addItem(roi)
+
+
+def test_getArrayRegion():
+    global vb, image, rois, data, size
+    
+    # Test we can call getArrayRegion without errors 
+    # (not checking for data validity)
+    for roi in rois:
+        arr = roi.getArrayRegion(data, image_tx)
+        assert arr.shape == size + data.shape[2:]
+        
+        arr = roi.getArrayRegion(data, image_tx, axes=(0, 1))
+        assert arr.shape == size + data.shape[2:]
+        
+        arr = roi.getArrayRegion(data.transpose(1, 0, 2, 3), image_tx, axes=(1, 0))
+        assert arr.shape == size + data.shape[2:]
+        
+        arr = roi.getArrayRegion(data, image_xy, axes=(1, 2))
+        assert arr.shape == data.shape[:1] + size + data.shape[3:]
+        
+        arr = roi.getArrayRegion(data.transpose(0, 2, 1, 3), image_xy, axes=(2, 1))
+        assert arr.shape == data.shape[:1] + size + data.shape[3:]
+        
+        arr, coords = roi.getArrayRegion(data, image_xy, axes=(1, 2), returnMappedCoords=True)
+        assert arr.shape == data.shape[:1] + size + data.shape[3:]
+        assert coords.shape == (2,) + size
+        
+        
+    
+    
\ No newline at end of file