fix this test

DeepRank · Oct 12, 2021 · e982adc · e982adc
1 parent 647222d
commit e982adc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_generate.py b/test/test_generate.py
@@ -13,6 +13,7 @@
 from deeprank.models.variant import PdbVariantSelection
 from deeprank.tools.sparse import FLANgrid
 from deeprank.operate import hdf5data
+from deeprank.domain.amino_acid import glycine, alanine
 
 
 _log = logging.getLogger(__name__)
@@ -40,7 +41,7 @@ def test_generate():
     target_names = ["test.target.target1"]
 
     pdb_path = "test/101m.pdb"
-    variants = [PdbVariantSelection(pdb_path, 'A', 25, 'G', 'A')]
+    variants = [PdbVariantSelection(pdb_path, 'A', 25, glycine, alanine)]
 
     tmp_dir = mkdtemp()
     try: