FIX: testios

LaurentRDC · Nov 7, 2018 · 794a6e1 · 794a6e1
1 parent d6d2718
commit 794a6e1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/crystals/base.py b/crystals/base.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
-from collections import Counter, OrderedDict
+from collections import Counter
+from collections import OrderedDict
 from itertools import chain
 
 import numpy as np
@@ -43,9 +44,9 @@ class AtomicStructure(Base):
         a secondary structure in a protein.
     """
 
-    def __init__(self, atoms = tuple(), substructures = tuple(), **kwargs):
-        self.atoms = frozenset(atoms)
-        self.substructures = frozenset(substructures)
+    def __init__(self, atoms = None, substructures = None, **kwargs):
+        self.atoms = frozenset(atoms or {})
+        self.substructures = frozenset(substructures or {})
         super().__init__(**kwargs)
 
     def __iter__(self):

diff --git a/tests/test_parsers.py b/tests/test_parsers.py
@@ -52,7 +52,7 @@ def test_symmetry_operators(self):
 
     def test_default_download_dir(self):
         """ Test that the file is saved in the correct temporary directory by default """
-        filename = PDBParser.retrieve_pdb_file('1fbb')
+        filename = PDBParser.download_pdb_file('1fbb')
 
         self.assertTrue(filename.exists())
         self.assertEqual(filename.parent, STRUCTURE_CACHE)