def testReadPlyCube(ply_name, expected_number_of_faces, expected_number_of_points): expected_cube_points = np.loadtxt(getTestFile("expected_" + ply_name + "_points.txt")) polyhedron = readPly(getTestFile(ply_name + ".ply")) assert len(polyhedron.getFaces()) == expected_number_of_faces assert len(polyhedron.getPoints()) == expected_number_of_points assert np.allclose(polyhedron.getPoints(), expected_cube_points)
def testMovePolyhedron(): expected_points = np.loadtxt(getTestFile("expected_moved_points.txt")) polyhedron = readPly(getTestFile('small_sphere.ply')) polyhedron.move(np.array([1.0, 2.0, 3.0])) assert np.allclose(polyhedron.getPoints(), expected_points)
def testRotatePolyhedron(): expected_points = np.loadtxt(getTestFile("expected_rotated_points.txt")) polyhedron = readPly(getTestFile('cube.ply')) polyhedron.rotate(np.array([45.0, 45.0, 45.0])) assert np.allclose(polyhedron.getPoints(), expected_points)
def createParticle(mass, color): particle = Particle(mass) particle.setGeometry(readPly('../objects/sphere.ply')) particle.geometry.scale(np.repeat(mass / 2.0, 3)) particle.geometry.setColor(color) return particle