feat: add Vector.from_numpy()

quantio/vector.py

@@ -47,6 +47,13 @@ def tile(cls, element: T | Vector[T] | list[T], length: int) -> Vector[T]:
 
         return Vector(np.tile(element, length))
 
+    @classmethod
+    def from_numpy(
+        cls, array: np.ndarray, element_class: type[_QuantityBase], unit: str
+    ) -> Vector[_QuantityBase]:
+        """Construct a quantity vector from a numpy array."""
+        return Vector([element_class(**{unit: elem}) for elem in array])
+
     def to_numpy(self, unit: str | None = None) -> np.ndarray[float]:
         """Convert this vector into a numpy array of floats."""
         if isinstance(self._elements[0], _QuantityBase):

test/test_vector.py

@@ -243,5 +243,11 @@ def test_sum__quantity():
     assert actual == Length(meters=6)
 
 
+def test_from_numpy__quantity():
+    array = np.array([0, 1, 2])
+    actual = Vector.from_numpy(array, Length, "meters")
+    assert actual == Vector([Length(meters=0), Length(meters=1), Length(meters=2)])
+
+
 if __name__ == "__main__":
     pytest.main([__file__, "-v"])