Method testCenterNthSelector

tests/test_selectors.py:226–352

(self)

                self.assertAlmostEqual(e.tangentAt(0).cross(vec).Length, 0.0)

    def testCenterNthSelector(self):
        sel = selectors.CenterNthSelector

        nothing = Workplane()
        self.assertEqual(nothing.solids().size(), 0)
        with self.assertRaises(ValueError):
            nothing.solids(sel(Vector(0, 0, 1), 0))

        c = Workplane(makeUnitCube(centered=True))

        bottom_face = c.faces(sel(Vector(0, 0, 1), 0))
        self.assertEqual(bottom_face.size(), 1)
        self.assertTupleAlmostEquals((0, 0, 0), bottom_face.val().Center().toTuple(), 3)

        side_faces = c.faces(sel(Vector(0, 0, 1), 1))
        self.assertEqual(side_faces.size(), 4)
        for f in side_faces.vals():
            self.assertAlmostEqual(0.5, f.Center().z)

        top_face = c.faces(sel(Vector(0, 0, 1), 2))
        self.assertEqual(top_face.size(), 1)
        self.assertTupleAlmostEquals((0, 0, 1), top_face.val().Center().toTuple(), 3)

        with self.assertRaises(IndexError):
            c.faces(sel(Vector(0, 0, 1), 3))

        left_face = c.faces(sel(Vector(1, 0, 0), 0))
        self.assertEqual(left_face.size(), 1)
        self.assertTupleAlmostEquals(
            (-0.5, 0, 0.5), left_face.val().Center().toTuple(), 3
        )

        middle_faces = c.faces(sel(Vector(1, 0, 0), 1))
        self.assertEqual(middle_faces.size(), 4)
        for f in middle_faces.vals():
            self.assertAlmostEqual(0, f.Center().x)

        right_face = c.faces(sel(Vector(1, 0, 0), 2))
        self.assertEqual(right_face.size(), 1)
        self.assertTupleAlmostEquals(
            (0.5, 0, 0.5), right_face.val().Center().toTuple(), 3
        )

        with self.assertRaises(IndexError):
            c.faces(sel(Vector(1, 0, 0), 3))

        # lower corner faces
        self.assertEqual(c.faces(sel(Vector(1, 1, 1), 0)).size(), 3)
        # upper corner faces
        self.assertEqual(c.faces(sel(Vector(1, 1, 1), 1)).size(), 3)
        with self.assertRaises(IndexError):
            c.faces(sel(Vector(1, 1, 1), 2))

        for idx, z_val in zip([0, 1, 2], [0, 0.5, 1]):
            edges = c.edges(sel(Vector(0, 0, 1), idx))
            self.assertEqual(edges.size(), 4)
            for e in edges.vals():
                self.assertAlmostEqual(z_val, e.Center().z)