switch to pytest

Author: pcmoritz

ci/travis_script_plasma.sh

@@ -81,8 +81,9 @@ python_version_tests() {
   popd
 
   # Run Plasma tests
-  pushd $TRAVIS_BUILD_DIR/cpp/src/plasma
-    python test/test.py
+  pushd $TRAVIS_BUILD_DIR/python
+    python -m pytest pyarrow/tests/test_plasma.py
+    VALGRIND=1 python -m pytest pyarrow/tests/test_plasma.py
   popd
 }
 

python/pyarrow/tests/conftest.py

@@ -18,11 +18,12 @@
 from pytest import skip
 
 
-groups = ['hdfs', 'parquet', 'large_memory']
+groups = ['hdfs', 'parquet', 'plasma', 'large_memory']
 
 defaults = {
     'hdfs': False,
     'parquet': False,
+    'plasma': False
     'large_memory': False
 }
 
@@ -32,6 +33,11 @@
 except ImportError:
     pass
 
+try:
+    import pyarrow.plasma as plasma
+    defaults['plasma'] = True
+except ImportError:
+    pass
 
 def pytest_configure(config):
     pass