fixed test generate

DeepRank · Apr 6, 2020 · 0810416 · 0810416
1 parent 1e61cda
commit 0810416
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/test/test_generate.py b/test/test_generate.py
@@ -1,7 +1,7 @@
 import os
 import unittest
 from time import time
-
+import shutil
 from deeprank.generate import *
 
 
@@ -247,7 +247,14 @@ def test_6_align_interface(self):
 
     def test_7_realign(self):
         '''Realign existing pdbs.'''
-        database = DataGenerator(hdf5='./1ak4.hdf5')
+
+        src_name = './1ak4.hdf5'
+        copy_name = './1ak4_aligned.hdf5'
+
+        os.remove(copy_name)
+        shutil.copy(src_name,copy_name)
+
+        database = DataGenerator(hdf5=copy_name)
         database.realign_complexes(align={'axis':'z'})