fix doctest on python3.7

ProteinQure · Feb 26, 2019 · d891351 · d891351
1 parent e2ccfd8
commit d891351
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mordred/_atomic_property.py b/mordred/_atomic_property.py
@@ -112,7 +112,8 @@ def map(self, f):
 _table = Chem.GetPeriodicTable()
 
 
-GetElementSymbol = _table.GetElementSymbol
+def GetElementSymbol(i):
+    return _table.GetElementSymbol(i)
 
 
 if six.PY2:
@@ -121,7 +122,8 @@ def GetAtomicNumber(symbol):
             symbol = str(symbol)
         return _table.GetAtomicNumber(symbol)
 else:
-    GetAtomicNumber = _table.GetAtomicNumber
+    def GetAtomicNumber(symbol):
+        return _table.GetAtomicNumber(symbol)
 
 
 # http://dx.doi.org/10.1002%2Fjps.2600721016