fix: module typing

engnadeau · Aug 23, 2022 · 8f652bf · 8f652bf
1 parent 85b997f
commit 8f652bf
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_tool.py b/tests/test_tool.py
@@ -9,11 +9,11 @@ def test_tool() -> None:
     """Test."""
     tool = Tool()
 
-    cg = [1, 2, 3]
+    cg = np.array([1, 2, 3])
     tool.cg = cg
-    np.testing.assert_allclose(tool.cg, cg)
+    np.testing.assert_allclose(tool.cg, cg)  # type: ignore
 
-    p = [1, 2, 3]
+    p = np.array([1, 2, 3])
     tool.position = p
-    np.testing.assert_allclose(tool.position, p)
-    np.testing.assert_allclose(tool.vector, matrix_2_vector(tool.matrix))
+    np.testing.assert_allclose(tool.position, p)  # type: ignore
+    np.testing.assert_allclose(tool.vector, matrix_2_vector(tool.matrix))  # type: ignore