7^(h)dZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZddlmZedd Zed Zed Zed ZGd dZGddZy)zFTests for the ``sympy.physics.mechanics.wrapping_geometry.py`` module.N) IntegerRationalSSymbolacoscospisinsqrt)Eq)PointReferenceFrameWrappingCylinderWrappingSpheredynamicsymbolssimplifyrTpositivexqNc( eZdZedZeej jdeje de ze jzgdZeej jde e jze e jzejfe e jze e jzej eze zfe e jze e j zee zfe e j ze e jzee zfe e jze edzej ze je jzzeddeze zfe edzej ze je jzze edzeddze je jze j(zze eed eddzzfgd Zeej jd e e jze e jze je jfe e jze e jze j e jfe e jzeddz e ze jzeddz e ze jzz e jeddz e jzeddz e jzzfe e jze dz e jzeddz e ze jzze jeddz e jzd e jzz fe e jzeddz e ze jzeddz e ze jzze jeddz e jzeddz e jzz fgd Zeej jde e jze eeze jze eeze jzzgdZeej jde e jze e jzfe e jze e jzfe eeze jze eeze jzze eeze jze eeze jzz fgdZy)TestWrappingSpherectdd}td}t||}t|tsJt |dsJ|j |k(sJt |dsJ|j |k(sJy)NrTrpOradiuspoint)rr r isinstancehasattrrr)rrspheres r/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sympy/physics/mechanics/tests/test_wrapping_geometry.pytest_valid_constructorz)TestWrappingSphere.test_valid_constructor#sw 3 & 4[2&&.111vx(((}}!!!vw'''||r!!!positioncHtdd}td}t||}td}|j||td}|j||d}t j t |5|j||dddy#1swYyxYw NrTrrp1p2z'point .* does not lie on the surface of)match)rr rset_pospytestraises ValueErrorgeodesic_length)r&rrr"r*r+ error_msgs r#1test_geodesic_length_point_not_on_surface_invalidzDTestWrappingSphere.test_geodesic_length_point_not_on_surface_invalid.s 3 & 4[2& 4[ 2x 4[ 2x > ]]:Y 7 +  " "2r * + + +s <BB!z position_1, position_2, expectedctdd}td}t||}td}|j||td}|j||t t |j |||sJyNrTrrr*r+)rr rr-rr r1) position_1 position_2expectedrrr"r*r+s r#test_geodesic_lengthz'TestWrappingSphere.test_geodesic_length>su" 3 & 4[2& 4[ 2z" 4[ 2z"611"b98DEEEr%z*position_1, position_2, vector_1, vector_2?ctdd}td}t||}td}|j||td}|j||||f} |j ||| k(sJyr9)rr rr-geodesic_end_vectors) r:r;vector_1vector_2rrr"r*r+r<s r#test_geodesic_end_vectorsz,TestWrappingSphere.test_geodesic_end_vectorsZsx6 3 & 4[2& 4[ 2z" 4[ 2z"h'**2r2h>>>r%c@tdd}td}t||}td}|j||td}|j||t j t 5|j||}dddy#1swYyxYwr9rr rr-r.r/r0r@)r&rrr"r*r+_s r#,test_geodesic_end_vectors_invalid_coincidentz?TestWrappingSphere.test_geodesic_end_vectors_invalid_coincidents 3 & 4[2& 4[ 2x 4[ 2x ]]: & 4++B3A 4 4 4 8BBzposition_1, position_2c@tdd}td}t||}td}|j||td}|j||t j t 5|j||}dddy#1swYyxYwr9rE)r:r;rrr"r*r+rFs r#8test_geodesic_end_vectors_invalid_diametrically_oppositezKTestWrappingSphere.test_geodesic_end_vectors_invalid_diametrically_opposites 3 & 4[2& 4[ 2z" 4[ 2z" ]]: & 4++B3A 4 4 4rHN)__name__ __module__ __qualname__ staticmethodr$r.mark parametrizerZerorrrrr3yHalfr r rzrr=rCrrr rGrJr%r#rr!s"" [[Z!&&'!*Q,qss2B)CD +E + [[* qssUAaccE166 " qssUAaccE166"9Q; ' qssUAqssdFBqD ! tVQqssUBqD ! qssUAd1gIaff$accACCi0(1a.2CA2E F$q' !&& !##),$q' (1a.(!##)acc/:$tAwx1~-..   F F [[4 Wa!##gqssACC ( Wqb133hacc *ACCQ A #d1gai!macc&99Q ACC$q'!)acc/1  ACCA d1gai!macc11Q ACC#)+  ACCQ A #d1gai!macc&99Q ACC$q'!)acc/1   2 ?34 ? [[ QSS!c!f*qss"QQZ!##%556 4  4 [[ Wqb133h R!##Xq133w CF QSS 1s1v:#33SV acc!AAJ$44   4 4r%rceZdZedZeej jdejdfe e jzdfe e jzdfe e je jzjzdfede ze jzdfe e j e jzzdfe ede j ze jzzdfede j ze ede jze jzjzzdfe eee jzeee jzzzdfg dZeej jdejede ze jzgdZeej jd e j e e jze e jzejfe j e e jze j e e jzzej,fe j e e jze e j ze e jzzeedzfe j e e jzee j ze e jzzeedzfe j e e jze e jzej0ezee dzzfe j e e jze e jzed ej0zezee dzzfe j e e jze e jzed ej0zezee dzzfe j e e jze e jzej0ezee dzzfe j e e jze eee jzeee jzzzee dzedzzfe j e e jze eee jzeee jzzzee dzedezez dzzfg d Zeej jd e je e j ze e jze je j fe je e j ze e j ze je jfe je e j ze e j ze j e j fe j e e j ze e j ze j e j fe j e e j ze e j ze je jfe je e j ze e jze je j fe je e jzeddz e ze j zeddz e ze jzz e j ed dz e j zeddz e jzz fe je e j ze dz e j zed dz e ze jzze jed dz e j zd e jzz fe je e j zeddz e ze j zeddz e ze jzze jeddz e j zeddz e jzz fe je e j ze e j ze jze je j fe je e j ze e jzedz e ze jzzeddz e jzeddz e jzzeddz e j zeddz e jzz fe je e j ze eeze j ze eeze jzze jeee j zeee jzz fg dZeej jde je e j zfe je eeze j ze eeze jzze jzfgdZy)TestWrappingCylindercjtd}tdd}td}t|||j}t |tsJt |dsJ|j|k(sJt |dsJ|j|k(sJt |dsJ|j|jk(sJy) NrrTrrrraxis) rrr rrr r!rrrY)rrrcylinders r#r$z+TestWrappingCylinder.test_valid_constructors 3  3 & 4[#Ar133/($4555x***!###x)))~~###x(((}}###r%zposition, expectedFTr'ctdd}td}t||tj}td}|j |||j ||usJy)NrTrrr*)rr rrrr-point_on_surface)r&r<rrrZr*s r#test_point_is_on_surfacez-TestWrappingCylinder.test_point_is_on_surfaces[ 3 & 4[#Ar133/ 4[ 2x ((,888r%r&cftdd}td}t||tj}td}|j ||td}|j ||d}t jt|5|j||dddy#1swYyxYwr)) rr rrrr-r.r/r0r1)r&rrrZr*r+r2s r#r3zFTestWrappingCylinder.test_geodesic_length_point_not_on_surface_invalids 3 & 4[#Ar133/ 4[ 2x 4[ 2x > ]]:Y 7 -  $ $R , - - -s B''B0z&axis, position_1, position_2, expectedr6ctdd}td}t|||}td}|j||td}|j||t t |j |||sJyr9)rr rr-rr r1) rYr:r;r<rrrZr*r+s r#r=z)TestWrappingCylinder.test_geodesic_lengthsw* 3 & 4[#Ar40 4[ 2z" 4[ 2z"833B;XFGGGr%z0axis, position_1, position_2, vector_1, vector_2r>ctdd}td}t|||}td}|j||td} | j||||f} t d|j || D} | | k(sJy)NrTrrr*r+c3<K|]}|jyw)Nr).0 end_vectors r# zATestWrappingCylinder.test_geodesic_end_vectors..Qs!     ! s)rr rr-tupler@) rYr:r;rArBrrrZr*r+r< end_vectorss r#rCz.TestWrappingCylinder.test_geodesic_end_vectors sz 3 & 4[#Ar40 4[ 2z" 4[ 2z"h' &;;BC  h&&&r%zaxis, positioncBtdd}td}t|||}td}|j||td}|j||t j t 5|j||}dddy#1swYyxYwr9)rr rr-r.r/r0r@)rYr&rrrZr*r+rFs r#rGzATestWrappingCylinder.test_geodesic_end_vectors_invalid_coincidentXs 3 & 4[#Ar40 4[ 2x 4[ 2x ]]: & 6--b"5A 6 6 6s 9BBN)rKrLrMrNr$r.rOrPrrQrrrRrT normalizerrrrr r]r3Oner rSr r=rCrGrUr%r#rWrWsv $ $ [[ VVUO qssUDM qssUDM acc $$& & - QZ\!## u % acc ]D !  133$ %t , QZ^aACC!##!5 @ @ BB BD I Aqss SVACCZ' ($ /  9 9 [[Z!&&'!*Q,qss2B)CD -E - [[0 SS!ACC%133 ' SS!ACC%quaee , SS!ACC%!ACC!ACC%ad 4ccT1QSS5!ACC%!ACC%-ad 4 SS!ACC%133r $q!t* 4 5ccT1QSS5!ACC%AFF!22!5d1a4j!@ A SS!ACC%133 166 1" 4T!Q$Z ? @ccT1QSS5!ACC%41:!5 6 SS!ACC%CF133JQ34d1a419o Fae3q6!##:Aqss *+QT71:b=1,q001  & H'( H [[: SS!acc'1qss7ACC - SS!acc'A28QSS!## . SS1"qss(AGaccTACC4 0ccT1qss7QBHqssdQSSD 1ccTA28QWacc133 / SS!acc'A28QSS133$ /ACCQ A #d1gai!macc&99##q' ! acc!DGAIO3  ACCA d1gai!macc11Q ACC#)+  ACCQ A #d1gai!macc&99Q ACC$q'!)acc/1  ACCACC!##   ACCACC"Q$(QSS.(Q ACC$q'!)acc/1Q ACC$q'!)acc/1  ACCCF QSS 1s1v:#33A s1v|+  U1 4j'k4l'0 [[ SS!acc'N SS!c!f*qss"QQZ!##%55; <  6 6r%rW)__doc__r.sympyrrrrrrr r r sympy.core.relationalr sympy.physics.mechanicsr rrrrsympy.simplify.simplifyrrrrrrrWrUr%r#rosrL    %- 3 3K33M4M4`z6z6r%