Add init and close pytests.

Author: sebpuetz

.travis.yml

@@ -21,13 +21,13 @@ install:
     if [ "$TRAVIS_OS_NAME" == "linux" ]; then
       python3.6 -m venv venv
       source venv/bin/activate
-      pip install tensorflow virtualenv
+      pip install tensorflow virtualenv pytest
     fi
   - |
     if [ "$TRAVIS_OS_NAME" == "osx" ]; then
       python3 -m venv venv
       source venv/bin/activate
-      pip install tensorflow virtualenv
+      pip install tensorflow virtualenv pytest
     fi
 
 script: ci/script.sh

CMakeLists.txt

@@ -14,3 +14,4 @@ add_definitions(-D_GLIBCXX_USE_CXX11_ABI=${CXX_11_ABI})
 
 add_subdirectory(finalfusion-cxx)
 add_subdirectory(finalfusion-tf)
+add_subdirectory(tests)

ci/script.sh

@@ -7,11 +7,4 @@ cd build
 cmake ..
 make
 
-# Verify that we can load the shared library in tensorflow.
-if [ "$TRAVIS_OS_NAME" == "osx" ]; then
-  python3 -c "import tensorflow as tf; tf.load_op_library('finalfusion-tf/libfinalfusion_tf.dylib')"
-fi
-
-if [ "$TRAVIS_OS_NAME" == "linux" ]; then
-  python3.6 -c "import tensorflow as tf; tf.load_op_library('finalfusion-tf/libfinalfusion_tf.so')"
-fi
+ctest

tests/CMakeLists.txt

@@ -0,0 +1,9 @@
+include(CTest)
+
+file(COPY testdata/test.fifu DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/data)
+message(${CMAKE_CURRENT_BINARY_DIR})
+
+add_test(NAME python-init-close
+        COMMAND pytest ${CMAKE_CURRENT_SOURCE_DIR}
+        WORKING_DIRECTORY ${PROJECT_BINARY_DIR}
+        )

tests/conftest.py

@@ -0,0 +1,22 @@
+import os
+import platform
+
+import pytest
+import tensorflow as tf
+
+tf.enable_eager_execution()
+
+
+@pytest.fixture
+def ff_lib(tests_root):
+    if platform.system() == "Darwin":
+        LIB_SUFFIX = ".dylib"
+    else:
+        LIB_SUFFIX = ".so"
+
+    yield tf.load_op_library("./finalfusion-tf/libfinalfusion_tf" + LIB_SUFFIX)
+
+
+@pytest.fixture
+def tests_root():
+    yield os.path.dirname(__file__)

tests/test_init_close.py

@@ -0,0 +1,14 @@
+import platform
+import pytest
+
+
+def test_init_and_close(ff_lib):
+    embeddings = ff_lib.ff_embeddings()
+    ff_lib.initialize_ff_embeddings(embeddings, "tests/data/test.fifu", mmap=False)
+    ff_lib.close_ff_embeddings(embeddings)
+
+
+def test_init_and_close_mmap(ff_lib):
+    embeddings = ff_lib.ff_embeddings()
+    ff_lib.initialize_ff_embeddings(embeddings, "tests/data/test.fifu", mmap=True)
+    ff_lib.close_ff_embeddings(embeddings)