Allow custom unit class to not be iterable

dstansby · dstansby · commit ca00ccaf196c · 2018-01-09T20:57:13.000Z
diff --git a/lib/matplotlib/tests/test_units.py b/lib/matplotlib/tests/test_units.py
@@ -31,7 +31,10 @@ def __getattr__(self, attr):
         return getattr(self.magnitude, attr)
 
     def __getitem__(self, item):
-        return Quantity(self.magnitude[item], self.units)
+        if iterable(self.magnitude):
+            return Quantity(self.magnitude[item], self.units)
+        else:
+            return Quantity(self.magnitude[item], self.units)
 
     def __array__(self):
         return np.asarray(self.magnitude)
