Added exception and unittest for dist_mat_to_vec of analysis/psa.py (…

…Part of MDAnalysis#597)
vedantrathore · Feb 3, 2017 · 36acf58 · 36acf58
1 parent ece534a
commit 36acf58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/package/MDAnalysis/analysis/psa.py b/package/MDAnalysis/analysis/psa.py
@@ -212,7 +212,7 @@
 from six.moves import range, cPickle
 
 import numpy as np
-import warnings
+import warnings,numbers
 
 import MDAnalysis
 import MDAnalysis.analysis.align
@@ -575,7 +575,7 @@ def dist_mat_to_vec(N, i, j):
       int, index (of the matrix element) in the corresponding distance vector
     """
 
-    if not (isinstance(N, int) or isinstance(i, int) or isinstance(j, int)):
+    if not (isinstance(N, numbers.Integral) or isinstance(i, numbers.Integral) or isinstance(j, numbers.Integral)):
         raise ValueError("N, i, j all must be of type int")
 
     if i < 0 or j < 0 or N < 2: