Make TaggedValue in basic_units a sequence

dstansby · dstansby · commit 1da66db63e72 · 2021-02-01T12:38:02.000Z
diff --git a/examples/units/basic_units.py b/examples/units/basic_units.py
@@ -142,6 +142,12 @@ def __getattribute__(self, name):
     def __array__(self, dtype=object):
         return np.asarray(self.value).astype(dtype)
 
+    def __len__(self):
+        return len(self.value)
+
+    def __getitem__(self, key):
+        return type(self)(self.value[key], self.unit)
+
     def __array_wrap__(self, array, context):
         return TaggedValue(array, self.unit)
 
