Deprecated to_axis Issue #155

tests/test_direct_api/test_axis.py

@@ -33,7 +33,7 @@ import unittest
 import numpy as np
 from OCP.gp import gp_Ax1, gp_Dir, gp_Pnt
 from build123d.geometry import Axis, Location, Plane, Vector
-from build123d.topology import Edge
+from build123d.topology import Edge, Vertex
 
 
 class AlwaysEqual:
@@ -65,10 +65,18 @@ class TestAxis(unittest.TestCase):
         self.assertAlmostEqual(test_axis.position, (1, 2, 3), 5)
         self.assertAlmostEqual(test_axis.direction, (0, 0, 1), 5)
 
+        with self.assertRaises(ValueError):
+            Axis("one")
         with self.assertRaises(ValueError):
             Axis("one", "up")
         with self.assertRaises(ValueError):
             Axis(one="up")
+        with self.assertRaises(ValueError):
+            bad_edge = Edge()
+            bad_edge.wrapped = Vertex(0, 1, 2).wrapped
+            Axis(edge=bad_edge)
+        with self.assertRaises(ValueError):
+            Axis(gp_ax1=Edge.make_line((0, 0), (1, 0)))
 
     def test_axis_from_occt(self):
         occt_axis = gp_Ax1(gp_Pnt(1, 1, 1), gp_Dir(0, 1, 0))
@@ -100,6 +108,11 @@ class TestAxis(unittest.TestCase):
         self.assertAlmostEqual(y_axis.position, (0, 0, 1), 5)
         self.assertAlmostEqual(y_axis.direction, (0, 1, 0), 5)
 
+    def test_from_location(self):
+        axis = Axis(Location((1, 2, 3), (-90, 0, 0)))
+        self.assertAlmostEqual(axis.position, (1, 2, 3), 6)
+        self.assertAlmostEqual(axis.direction, (0, 1, 0), 6)
+
     def test_axis_to_plane(self):
         x_plane = Axis.X.to_plane()
         self.assertTrue(isinstance(x_plane, Plane))