U L?h)@sdZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZddlmZeddd Zed Zed Zed ZGd ddZGdddZdS)zFTests for the ``sympy.physics.mechanics.wrapping_geometry.py`` module.N) IntegerRationalSSymbolacoscospisinsqrt)Eq)PointReferenceFrameWrappingCylinderWrappingSpheredynamicsymbolssimplifyrTZpositivexqNc@sxeZdZeddZeejdej e de e j gddZeejde e j e e j ej fe e j e e jejee fe e j e e j ee fe e j e e j ee fe e j e edeje j e jedd ee fe edeje j e je ed edd e j e je je eed edd fgd d Zeejde e j e e je je j fe e j e e je j e j fe e jedde e j edde e je j edde j edde jfe e j e de j ed de e je jed de j de jfe e j edde e j edde e je jedde j edde jfgddZeejde e j e eee j e eee jgddZeejde e j e e j fe e je e jfe eee j e eee je eee j e eee jfgddZdS)TestWrappingSpherecCshtddd}td}t||}t|ts,tt|ds:t|j|ksHtt|dsVt|j|ksdtdS)NrTrpOradiuspoint)rr r isinstanceAssertionErrorhasattrrr)rrspherer f/var/www/html/venv/lib/python3.8/site-packages/sympy/physics/mechanics/tests/test_wrapping_geometry.pytest_valid_constructor#s  z)TestWrappingSphere.test_valid_constructorpositionc Csttddd}td}t||}td}|||td}|||d}tjt|d|||W5QRXdS NrTrrp1p2z'point .* does not lie on the surface of)match)rr rset_pospytestraises ValueErrorgeodesic_length)r#rrrr&r' error_msgr r r!1test_geodesic_length_point_not_on_surface_invalid.s    zDTestWrappingSphere.test_geodesic_length_point_not_on_surface_invalidz position_1, position_2, expectedcCsdtddd}td}t||}td}|||td}|||tt||||s`tdSNrTrrr&r')rr rr)rr r-r) position_1 position_2expectedrrrr&r'r r r!test_geodesic_length>s    z'TestWrappingSphere.test_geodesic_lengthz*position_1, position_2, vector_1, vector_2?c Csftddd}td}t||}td}|||td}|||||f} |||| ksbtdSr4)rr rr)geodesic_end_vectorsr) r5r6vector_1vector_2rrrr&r'r7r r r!test_geodesic_end_vectorsZs    z,TestWrappingSphere.test_geodesic_end_vectorsc Csltddd}td}t||}td}|||td}|||tt|||}W5QRXdSr4rr rr)r*r+r,r:)r#rrrr&r'_r r r!,test_geodesic_end_vectors_invalid_coincidents     z?TestWrappingSphere.test_geodesic_end_vectors_invalid_coincidentzposition_1, position_2c Csltddd}td}t||}td}|||td}|||tt|||}W5QRXdSr4r>)r5r6rrrr&r'r?r r r!8test_geodesic_end_vectors_invalid_diametrically_opposites     zKTestWrappingSphere.test_geodesic_end_vectors_invalid_diametrically_oppositeN)__name__ __module__ __qualname__ staticmethodr"r*mark parametrizerZerorrrrr/yHalfrr rzrr8r=rrr r@rAr r r r!r!sx   6& *""*" , "$ rc@s@eZdZeddZeejdej dfe e j dfe e j dfe e j e j dfede e j dfe e je j dfe ede je j dfede je ede j e j dfe eee j eee j dfg ddZeejd ej ede e j gd d Zeejd e je e j e e j ej fe je e j e je e j ejfe je e j e e je e j eedfe j e e j ee je e j eedfe je e j e e j ejeee dfe j e e j e e j ed ejeee dfe je e j e e j ed ejeee dfe j e e j e e j ejeee dfe je e j e eee j eee j ee dedfe j e e j e eee j eee j ee dedeedfg ddZeejde j e e je e j e j e jfe j e e je e je j e j fe j e e je e je j e j fe j e e je e je j e j fe j e e je e je j e j fe j e e je e j e j e j fe j e e j edde e jedde e j e j ed de jedde j fe j e e je de jed de e j e j ed de jde j fe j e e jedde e jedde e j e j edde jedde j fe j e e je e je j e j e j fe j e e je e j ede e j edde j edde j edde jedde j fe j e e je eee je eee j e j eee jeee j fg ddZeejde j e e jfe j e eee je eee j e j fgddZdS)TestWrappingCylindercCstd}tddd}td}t|||j}t|ts8tt|dsFt|j|ksTtt|dsbt|j |ksptt|ds~t|j |jkstdS) NrrTrrrraxis) r rr rrrrrrrrM)rrrcylinderr r r!r"s z+TestWrappingCylinder.test_valid_constructorzposition, expectedFTr$cCsLtddd}td}t||tj}td}||||||ksHtdS)NrTrrr&)rr rrrr)Zpoint_on_surfacer)r#r7rrrNr&r r r!test_point_is_on_surfaces   z-TestWrappingCylinder.test_point_is_on_surfacer#c Csxtddd}td}t||tj}td}|||td}|||d}tjt|d| ||W5QRXdSr%) rr rrrr)r*r+r,r-)r#rrrNr&r'r.r r r!r/s   zFTestWrappingCylinder.test_geodesic_length_point_not_on_surface_invalidz&axis, position_1, position_2, expectedr2c Csftddd}td}t|||}td}|||td}|||tt||||sbtdSr4)rr rr)rr r-r) rMr5r6r7rrrNr&r'r r r!r8s    z)TestWrappingCylinder.test_geodesic_lengthz0axis, position_1, position_2, vector_1, vector_2r9c Csztddd}td}t|||}td}|||td} | ||||f} tdd||| D} | | ksvtdS) NrTrrr&r'css|]}|VqdS)Nr).0Z end_vectorr r r! QszATestWrappingCylinder.test_geodesic_end_vectors..)rr rr)tupler:r) rMr5r6r;r<rrrNr&r'r7Z end_vectorsr r r!r= s=     z.TestWrappingCylinder.test_geodesic_end_vectorszaxis, positionc Csntddd}td}t|||}td}|||td}|||tt|||}W5QRXdSr4)rr rr)r*r+r,r:)rMr#rrrNr&r'r?r r r!r@Xs     zATestWrappingCylinder.test_geodesic_end_vectors_invalid_coincidentN)rBrCrDrEr"r*rFrGrrHrrrIrK normalizerrrrr rOr/ZOner rJrr8r=r@r r r r!rLs   ,"    ,,*42,>  $&""*$"*""""5 .rL)__doc__r*Zsympyrrrrrrrr r Zsympy.core.relationalr Zsympy.physics.mechanicsr r rrrZsympy.simplify.simplifyrrrrrrrLr r r r!s,