Qi)dZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZddlmZedd Zed Zed Zed ZGd dZGddZdS)zFTests for the ``sympy.physics.mechanics.wrapping_geometry.py`` module.N) IntegerRationalSSymbolacoscospisinsqrt)Eq)PointReferenceFrameWrappingCylinderWrappingSpheredynamicsymbolssimplifyrTpositivexqNceZdZedZeejdej e de ze j zgdZeejde e j ze e j zej fe e j ze e jzejeze zfe e j ze e j zee zfe e j ze e j zee zfe e j ze edzejze j e jzzeddeze zfe edzejze j e jzze edzeddze j e jze jzze eed eddzzfgd Zeejd e e j ze e jze je j fe e j ze e jze j e j fe e jzeddz e ze j zeddz e ze jzz e j eddz e j zeddz e jzzfe e j ze dz e j zeddz e ze jzze jeddz e j zd e jzz fe e j zeddz e ze j zeddz e ze jzze jeddz e j zeddz e jzz fgd Zeejde e j ze eeze j ze eeze jzzgdZeejde e j ze e j zfe e jze e jzfe eeze j ze eeze jzze eeze j ze eeze jzz fgdZdS)TestWrappingSpherectdd}td}t||}t|tsJt |dsJ|j|ksJt |dsJ|j|ksJdS)NrTrpOradiuspoint)rr r isinstancehasattrrr)rrspheres /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/sympy/physics/mechanics/tests/test_wrapping_geometry.pytest_valid_constructorz)TestWrappingSphere.test_valid_constructor#s 3 & & & 4[[2&&&.11111vx(((((}!!!!vw'''''|r!!!!!!positionctdd}td}t||}td}|||td}|||d}t jt |5|||ddddS#1swxYwYdS 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 7 7 + +  " "2r * * * + + + + + + + + + + + + + + + + + +sB<<CCz position_1, position_2, expectedc`tdd}td}t||}td}|||td}|||t t ||||sJdSNrTrrr*r+)rr rr-rr r1) position_1 position_2expectedrrr"r*r+s r#test_geodesic_lengthz'TestWrappingSphere.test_geodesic_length>s" 3 & & & 4[[2&& 4[[ 2z""" 4[[ 2z"""611"b998DDEEEEEEEr%z*position_1, position_2, vector_1, vector_2?c:tdd}td}t||}td}|||td}|||||f} |||| ksJdSr9)rr rr-geodesic_end_vectors) r:r;vector_1vector_2rrr"r*r+r<s r#test_geodesic_end_vectorsz,TestWrappingSphere.test_geodesic_end_vectorsZs6 3 & & & 4[[2&& 4[[ 2z""" 4[[ 2z"""h'**2r22h>>>>>>r%ctdd}td}t||}td}|||td}|||t jt 5|||}ddddS#1swxYwYdSr9rr 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 4++B33A 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4B88B<?B<zposition_1, position_2ctdd}td}t||}td}|||td}|||t jt 5|||}ddddS#1swxYwYdSr9rE)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 4++B33A 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4rHN)__name__ __module__ __qualname__ staticmethodr$r.mark parametrizerZerorrrrr3yHalfr r rzrr=rCrrr rGrJr%r#rr!s ""\" [Z!&''!**Q,qs2B)CDD + +ED\ + [* qsUAacE16 " qsUAacE16"9Q; ' qsUAqsdFBqD ! tVQqsUBqD ! qsUAdd1ggIaf$acACi0((1a..2CA2E F$$q'' !& !#),$$q'' ((1a..(!#)ac/:$$ttAwwxx1~~-...   F F\ F [4 Wa!#gqsAC ( Wqb13hac *ACQ A #dd1ggai!mac&99Q AC$$q''!)ac/1  ACA dd1ggai!mac11Q AC#)+  ACQ A #dd1ggai!mac&99Q AC$$q''!)ac/1   2 ? ?3\4 ? [ QS!cc!ff*qs"QQZ!#%556 4 4 \ 4 [ Wqb13h R!#Xq13w CCFF QS 1ss1vv:#33SSVV ac!AAJ$44    44  \444r%rceZdZedZeejdej dfe e j zdfe e j zdfe e j e j zzdfede ze j zdfe e je j zzdfe ede jze j zzdfede jze ede j ze j zzzdfe eee j zeee j zzzdfg dZeejdej ede ze j zgdZeejd e je e j ze e j zej fe je e j ze je e j zzejfe je e j ze e jze e j zzeedzfe j e e j zee jze e j zzeedzfe je e j ze e j zejezee dzzfe j e e j ze e j zed ejzezee dzzfe je e j ze e j zed ejzezee dzzfe j e e j ze e j zejezee dzzfe je e j ze eee j zeee j zzzee dzedzzfe j e e j ze eee j zeee j zzzee dzedezez dzzfg d Zeejd e j e e jze e j ze j e jfe j e e jze e jze j e j fe j e e jze e jze j e j fe j e e jze e jze j e j fe j e e jze e jze j e j fe j e e jze e j ze j e j fe j e e j zeddz e ze jzeddz e ze j zz e j ed dz e jzeddz e j zz fe j e e jze dz e jzed dz e ze j zze j ed dz e jzd e j zz fe j e e jzeddz e ze jzeddz e ze j zze j eddz e jzeddz e j zz fe j e e jze e jze j ze j e j fe j e e jze e j zedz e ze j zzeddz e j zeddz e j zzeddz e jzeddz e j zz fe j e e jze eeze jze eeze j zze j eee jzeee j zz fg dZeejde j e e jzfe j e eeze jze eeze j zze j zfgdZdS)TestWrappingCylinderctd}tdd}td}t|||j}t |tsJt |dsJ|j|ksJt |dsJ|j|ksJt |dsJ|j |jksJdS) NrrTrrrraxis) rrr rrr r!rrrY)rrrcylinders r#r$z+TestWrappingCylinder.test_valid_constructors 3   3 & & & 4[[#Ar13//($455555x*****!####x)))))~####x(((((}######r%zposition, expectedFTr'ctdd}td}t||tj}td}||||||usJdS)NrTrrr*)rr rrrr-point_on_surface)r&r<rrrZr*s r#test_point_is_on_surfacez-TestWrappingCylinder.test_point_is_on_surfacesx 3 & & & 4[[#Ar13// 4[[ 2x   ((,,888888r%r&ctdd}td}t||tj}td}|||td}|||d}t jt|5| ||ddddS#1swxYwYdSr)) rr rrrr-r.r/r0r1)r&rrrZr*r+r2s r#r3zFTestWrappingCylinder.test_geodesic_length_point_not_on_surface_invalids 3 & & & 4[[#Ar13// 4[[ 2x   4[[ 2x   > ]:Y 7 7 7 - -  $ $R , , , - - - - - - - - - - - - - - - - - -s#CC C z&axis, position_1, position_2, expectedr6cbtdd}td}t|||}td}|||td}|||t t ||||sJdSr9)rr rr-rr r1) rYr:r;r<rrrZr*r+s r#r=z)TestWrappingCylinder.test_geodesic_lengths* 3 & & & 4[[#Ar400 4[[ 2z""" 4[[ 2z"""833B;;XFFGGGGGGGr%z0axis, position_1, position_2, vector_1, vector_2r>cntdd}td}t|||}td}|||td} | ||||f} t d||| D} | | ksJdS)NrTrrr*r+c3>K|]}|VdS)Nr).0 end_vectors r# zATestWrappingCylinder.test_geodesic_end_vectors..QsB      ! !      r%)rr rr-tupler@) rYr:r;rArBrrrZr*r+r< end_vectorss r#rCz.TestWrappingCylinder.test_geodesic_end_vectors sz 3 & & & 4[[#Ar400 4[[ 2z""" 4[[ 2z"""h'  &;;BCC     h&&&&&&r%zaxis, positionctdd}td}t|||}td}|||td}|||t jt 5|||}ddddS#1swxYwYdSr9)rr rr-r.r/r0r@)rYr&rrrZr*r+rFs r#rGzATestWrappingCylinder.test_geodesic_end_vectors_invalid_coincidentXs 3 & & & 4[[#Ar400 4[[ 2x   4[[ 2x   ]: & & 6 6--b"55A 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6sB99B=B=N)rKrLrMrNr$r.rOrPrrQrrrRrT normalizerrrrr r]r3Oner rSr r=rCrGrUr%r#rWrWsv $ $\ $ [ VUO qsUDM qsUDM ac $$&& & - WQZZ\!# u % ac ]D !  13$ %t , WQZZ^aAC!#!5 @ @ B BB BD I Aqs SSVVACZ' ($ /   99  \9 [Z!&''!**Q,qs2B)CDD - -ED\ - [0 S!AC%13 ' S!AC%quae , S!AC%!AC!AC%ad 4cT1QS5!AC%!AC%-ad 4 S!AC%13r $$q!t** 4 5cT1QS5!AC%AF!22!5dd1a4jj!@ A S!AC%13 16 1" 4TT!Q$ZZ ? @cT1QS5!AC%441::!5 6 S!AC%CCFF13JQ34dd1a419oo Fae33q66!#:Aqs *+QT771::b=1,q0011  & H H'\( H [: S!ac'1qs7AC - S!ac'A28QS!# . S1"qs(AGacTAC4 0cT1qs7QBHqsdQSD 1cTA28QWac13 / S!ac'A28QS13$ /ACQ A #dd1ggai!mac&99#$q'' ! ac!DDGGAIO3  ACA dd1ggai!mac11Q AC#)+  ACQ A #dd1ggai!mac&99Q AC$$q''!)ac/1  ACAC!#   ACAC"Q$(QS.(Q AC$$q''!)ac/1Q AC$$q''!)ac/1  ACCCFF QS 1ss1vv:#33A ss1vv|+  U1 44j''k44\l'0 [ S!ac'N S!cc!ff*qs"QQZ!#%55; <  6 6\ 6 6 6r%rW)__doc__r.sympyrrrrrrr r r sympy.core.relationalr sympy.physics.mechanicsr rrrrsympy.simplify.simplifyrrrrrrrWrUr%r#rosLL                       %$$$$$-,,,,, F3 F3KKN3N3M4M4M4M4M4M4M4M4`z6z6z6z6z6z6z6z6z6z6r%