Skip to content
This repository has been archived by the owner on Oct 8, 2024. It is now read-only.

Commit

Permalink
fix: argument typing
Browse files Browse the repository at this point in the history
  • Loading branch information
engnadeau committed Aug 23, 2022
1 parent 733ada5 commit 0922ed7
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions tests/test_geometry.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ def test_rotation_matrix_xyz(angle: float, resources_path: Path) -> None:
elements=st.floats(allow_nan=False, allow_infinity=False),
)
)
def test_translation_matrix(xyz) -> None:
def test_translation_matrix(xyz: npt.NDArray[np.float64]) -> None:
"""Test."""
matrix = pybotics.geometry.translation_matrix(xyz)

Expand All @@ -127,7 +127,9 @@ def test_translation_matrix(xyz) -> None:
pybotics.geometry.translation_matrix(np.zeros(10))


def test_vector_2_matrix(vector_transforms: Sequence[dict]) -> None:
def test_vector_2_matrix(
vector_transforms: Sequence[typing.Dict[str, npt.NDArray[np.float64]]]
) -> None:
"""Test."""
# test regular usage
for d in vector_transforms:
Expand All @@ -142,7 +144,9 @@ def test_vector_2_matrix(vector_transforms: Sequence[dict]) -> None:
pybotics.geometry.vector_2_matrix(d["vector"], convention="foobar")


def test_matrix_2_vector(vector_transforms: Sequence[dict]) -> None:
def test_matrix_2_vector(
vector_transforms: Sequence[typing.Dict[str, npt.NDArray[np.float64]]]
) -> None:
"""Test."""
for d in vector_transforms:
for c in [
Expand Down

0 comments on commit 0922ed7

Please # to comment.