Merge pull request apache#2 from dato-code/fix_tests

fix a few unittests

tests/python/gpu/test_conv.py

@@ -45,7 +45,7 @@ def get_iter(data_dir):
 logging.basicConfig(level=logging.DEBUG)
 
 num_gpus = 1
-data_dir = 's3://dmcl/mnist'
+data_dir = 'data'
 
 (train, val) = get_iter(data_dir)
 

tests/python/gpu/test_operator_gpu.py

@@ -1,7 +1,10 @@
+import os
 import sys
-sys.path.insert(0, '../unittest')
+curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
+sys.path.insert(0, os.path.join(curr_path, '..', 'unittest'))
+print sys.path
 from test_operator import *
 
 if __name__ == '__main__':
-	test_softmax_with_shape((3,4), mx.gpu())
-    test_multi_softmax_with_shape((3,4,5), mx.gpu())
+    test_softmax_with_shape((3,4), mx.gpu())
+    test_multi_softmax_with_shape((3,4,5), mx.gpu())

tests/python/train/test_conv.py

@@ -1,6 +1,4 @@
 # pylint: skip-file
-import sys
-sys.path.insert(0, '../../python')
 import mxnet as mx
 import numpy as np
 import os, pickle, gzip

tests/python/train/test_mlp.py

@@ -1,7 +1,7 @@
 # pylint: skip-file
 import mxnet as mx
 import numpy as np
-import os, sys
+import os
 import pickle as pickle
 import logging
 from common import get_data