From b0e42ab5f889233e9f22d0f22dde33cb329e103d Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Sat, 6 Feb 2021 16:42:14 +0100
Subject: [PATCH] [test] fix import

---
 nixview/test/create_test_file.py  | 12 +++++++-----
 nixview/test/test_file_handler.py |  2 +-
 nixview/test/test_ui.py           |  2 +-
 3 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/nixview/test/create_test_file.py b/nixview/test/create_test_file.py
index b589d9a..addb313 100644
--- a/nixview/test/create_test_file.py
+++ b/nixview/test/create_test_file.py
@@ -1,10 +1,9 @@
 import nixio as nix
-import lif
+import os
+import nixview.test.lif as lif
 import numpy as np
-import matplotlib.pyplot as plt
 import matplotlib.mlab as mlab
 from PIL import Image as img
-from IPython import embed
 
 def bivariate_normal(X, Y, sigmax=1.0, sigmay=1.0,
                      mux=0.0, muy=0.0, sigmaxy=0.0):
@@ -70,6 +69,7 @@ def create_1d_sampled(f, b):
 def create_2d(f, b, trials=10):
     # create multiple responses of a lif model neuron
     voltages = []
+    time = None
     for t in range(trials):
         time, voltage, _ = fake_neuron()
         voltages.append(voltage)
@@ -78,7 +78,7 @@ def create_2d(f, b, trials=10):
 
     da = b.create_data_array("membrane voltages", "nix.regular_sampled.series",
                              dtype=nix.DataType.Double, data=voltages)
-    d = da.append_sampled_dimension(time[1]-time[0])
+    d = da.append_sampled_dimension(time[1] - time[0])
     d.label = "time"
     d.unit = "s"
     da.label = "voltage"
@@ -103,7 +103,9 @@ def create_2d(f, b, trials=10):
 
 def create_3d(f, b):
     # taken from nix tutorial
-    image = img.open('lena.bmp')
+    path = os.path.realpath(__file__).split(os.sep)[:-1]
+    path.append("lena.bmp")
+    image = img.open(os.sep.join(path))
     img_data = np.array(image)
     channels = list(image.mode)
     data = b.create_data_array("lena", "nix.image.rgb", data=img_data)
diff --git a/nixview/test/test_file_handler.py b/nixview/test/test_file_handler.py
index 255e7ae..8a43e49 100644
--- a/nixview/test/test_file_handler.py
+++ b/nixview/test/test_file_handler.py
@@ -8,7 +8,7 @@ test_file = os.path.join(".", "nix_test.h5")
 def setup():
     if not os.path.exists(test_file):
         from nixview.test.create_test_file import create_test_file
-        create_test_file()
+        create_test_file(test_file)
     fhandler = file_handler.FileHandler()
     yield fhandler
     fhandler.close()
diff --git a/nixview/test/test_ui.py b/nixview/test/test_ui.py
index 73323ce..e46da76 100644
--- a/nixview/test/test_ui.py
+++ b/nixview/test/test_ui.py
@@ -9,7 +9,7 @@ test_file = os.path.join(".", "nix_test.h5")
 def app(qtbot):
     if not os.path.exists(test_file):
         from nixview.test.create_test_file import create_test_file
-        create_test_file()
+        create_test_file(test_file)
     test_nv = NixView()
     qtbot.addWidget(test_nv)
     yield test_nv