Compare sizes instead of shapes for dimensionality.

lewisamarshall · Jan 4, 2017 · 6f43aa0 · 6f43aa0
1 parent 7020ae1
commit 6f43aa0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ionize/Ion/__init__.py b/ionize/Ion/__init__.py
@@ -118,8 +118,10 @@ def __init__(self, name, valence, reference_pKa, reference_mobility,
                      'enthalpy',
                      'heat_capacity'):
             if getattr(self, prop) is not None:
-                assert getattr(self, prop).shape == self.valence.shape, \
-                    '{} must have the same shape as valence.'.format(prop)
+                template = 'In {}, valence is {}, while {} is {}. Wrong size.'
+                assert getattr(self, prop).size == self.valence.size, \
+                    template.format(self.name, self.valence, prop,
+                                    getattr(self, prop))
 
         assert np.all((self.reference_mobility / self.valence) > 0.), \
             'Mobilities must be signed. {}, {}'.format(self.reference_mobility,

diff --git a/ionize/__version__.py b/ionize/__version__.py
@@ -1 +1 @@
-__version__ = '1.1.2'
+__version__ = '1.1.3'