Organize the test code.

test/README.rst

@@ -2,6 +2,8 @@ The test utilize docker to run test automatically. The test script will start a
 
 The test scripts should be ended by a suffix '_test.py'
 
+Run :code:`simple_test.py` to make the downloaded binary can work correctly.
+
 1. Install dependencies first
 
 :code:`pip install -r requirements.txt`

test/conftest.py

@@ -34,3 +34,19 @@ def iserror(res):
 
 def isok(res):
     return res == 'ok'
+
+class Version:
+    '''
+    Represent the version information, can be used for comparison
+    '''
+    def __init__(self, data):
+        if isinstance(data, str):
+            # parse vx.y.z
+            [self.x, self.y, self.z] = [int(v) for v in data.lstrip('v').split('.')]
+
+    def __cmp__(self, v):
+        if self.x != v.x:
+            return cmp(self.x, v.x)
+        if self.y != v.y:
+            return cmp(self.y, v.y)
+        return cmp(self.z, v.z)

test/simple_test.py

@@ -0,0 +1,14 @@
+
+from unrealcv import client
+from conftest import env
+
+def test_gt(env):
+    client.connect()
+    gt_types = ['lit', 'depth', 'normal', 'object_mask']
+    for v in gt_types:
+        res = client.request('vget /camera/0/{type} output/{type}.png'.format(type=v))
+        print(res)
+        im = cv2.imread(res)
+        print(im.shape)
+
+    # Make sure the dimension is right

test/stereo_test.py

@@ -1,7 +1,7 @@
 # pytest -s stereo.py -k [name]
 from unrealcv import client
 import math, random
-from common import iserror, isok
+from conftest import iserror, isok, Version
 
 def get_version():
     res = client.request('vget /unrealcv/version')
@@ -10,19 +10,6 @@ def get_version():
     else:
         return res
 
-class Version:
-    def __init__(self, data):
-        if isinstance(data, str):
-            # parse vx.y.z
-            [self.x, self.y, self.z] = [int(v) for v in data.lstrip('v').split('.')]
-
-    def __cmp__(self, v):
-        if self.x != v.x:
-            return cmp(self.x, v.x)
-        if self.y != v.y:
-            return cmp(self.y, v.y)
-        return cmp(self.z, v.z)
-
 class Vec3:
     def __init__(self, data):
         if isinstance(data, str):