X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fcreate_pickle_file.py;h=63d139674d4d1571f63c826fdaa02f10f20726f2;hb=7c5adcf962fe99d6a42307595f263f47d0112fbc;hp=55843a910b2ba3f8ffb348e587ab189269d6bb24;hpb=b9d3d4be0a931b00a4d96b39ce61e1a374870ad3;p=dak.git

diff --git a/tests/create_pickle_file.py b/tests/create_pickle_file.py
index 55843a91..63d13967 100755
--- a/tests/create_pickle_file.py
+++ b/tests/create_pickle_file.py
@@ -3,14 +3,20 @@
 # recreate the pickle file db-metadata-*.pkl that needs to be updated
 # after a database upgrade
 
-from base_test import fixture
-from daklib.dbconn import DBConn
-
 from sqlalchemy import create_engine, __version__
 
 import pickle
+import sys
+from os.path import abspath, dirname
+
+DAK_TEST_DIR = dirname(abspath(__file__))
+DAK_ROOT_DIR = dirname(DAK_TEST_DIR)
+if DAK_ROOT_DIR not in sys.path:
+    sys.path.insert(0, DAK_ROOT_DIR)
+
+from daklib.dbconn import DBConn
 
-pickle_filename = fixture('db-metadata-%s.pkl' % __version__)
+pickle_filename = '%s/fixtures/db-metadata-%s.pkl' % (DAK_TEST_DIR, __version__)
 pickle_file = open(pickle_filename, 'w')
 metadata = DBConn().db_meta
 pickle.dump(metadata, pickle_file)