a k║Фhpaу@sвdZddlZddlmZmZmZmZmZmZddl m Z mZmZm Z mZmZmZmZmZmZmZddlmZddДZGdd Дd ГZGd dДdГZGdd Дd ГZdS)zs z#_simplify_loads..r)┌loadsrrr┌_simplify_loadss■rc @seZdZddДZeejаdedГedГfifgбddДГГZ eejаdedГfedГedГed Гfgбd dДГГZ eejаddedГfedГdfgбd dДГГZejddНddДГZ ddДZddДZddДZddДZddДZddДZdd ДZd!d"ДZd#d$ДZd%d&ДZd'd(ДZdS))┌TestLinearPathwaycCstttГsJВdSйN)┌ issubclassr rй┌selfrrr┌test_is_pathway_base_subclass&sz/TestLinearPathway.test_is_pathway_base_subclasszargs, kwargs┌pA┌pBcCs╝|\}}t|i|дО}t|tГs$JВt|dГs2JВt|jГdksDJВ|jd|usVJВ|jd|ushJВt|jdtГs|JВ|jdjdksРJВt|jdtГsдJВ|jdjdks╕JВdS)N┌attachmentsщrщr#r$)r ┌ isinstance┌hasattr┌lenr%r┌name)┌args┌kwargsZpointAZpointB┌instancerrr┌test_valid_constructor)sz(TestLinearPathway.test_valid_constructorr%┌pZcCs6tаtбПt|О}WdГn1s(0YdSr)┌pytest┌raises┌ ValueErrorr йr%┌_rrr┌)test_invalid_attachments_incorrect_number=s z;TestLinearPathway.test_invalid_attachments_incorrect_numberNcCs6tаtбПt|О}WdГn1s(0YdSr)r1r2┌ TypeErrorr r4rrr┌"test_invalid_attachments_not_pointIs z4TestLinearPathway.test_invalid_attachments_not_pointTйZautousecCs~tdГ|_tdГ|_tdГ|_t|j|jГ|_tdГ|_tdГ|_ tdГ|_ tddГ|_tddГ|_tddГ|_ tdГ|_dS) N┌Nr#r$┌q1┌q2┌q3r'┌F)r r:rr#r$r ┌pathwayrr;r<r=┌q1d┌q2d┌q3drr>r rrr┌_linear_pathway_fixtureUs z)TestLinearPathway._linear_pathway_fixturecCsкt|j|jГ}tаtбПd|_WdГn1s40YtаtбПd|jd<WdГn1sh0YtаtбПd|jd<WdГn1sЬ0YdSйNrr')r r#r$r1r2┌AttributeErrorr%r7йr!r.rrr┌test_properties_are_immutablecs$(z/TestLinearPathway.test_properties_are_immutablecCs&t|j|jГ}d}t|Г|ks"JВdS)NzLinearPathway(pA, pB))r r#r$┌reprйr!r?┌expectedrrr┌ test_reprlszTestLinearPathway.test_reprcCs,|jа|jd|jjб|jjdks(JВdSйNr&)r$┌set_posr#r:┌xr?┌lengthr rrr┌test_static_pathway_lengthqsz,TestLinearPathway.test_static_pathway_lengthcCs,|jа|jd|jjб|jjdks(JВdSйNr&r)r$rMr#r:rNr?┌extension_velocityr rrr┌&test_static_pathway_extension_velocityusz8TestLinearPathway.test_static_pathway_extension_velocitycCs\|jа|jd|jjб|j|j|jjf|j|j|jjfg}|jа|jб|ksXJВdSrL)r$rMr#r:rNr>r?┌to_loadsйr!rJrrr┌test_static_pathway_to_loadsys ■z.TestLinearPathway.test_static_pathway_to_loadscCsD|jа|jd|j|jjбdt|jdГ}|jj|ks@JВdSrL) r$rMr#r;r:rNrr?rOrUrrr┌test_2D_pathway_lengthБsz(TestLinearPathway.test_2D_pathway_lengthcCsP|jа|jd|j|jjбdt|jdГ|j|j}|jj |ksLJВdSrL) r$rMr#r;r:rNrr@r?rRrUrrr┌"test_2D_pathway_extension_velocityЖsz4TestLinearPathway.test_2D_pathway_extension_velocitycCsК|jа|jd|j|jjб|j|j|jt|jdГ|jjf|j|j|jt|jdГ|jjfg}|jа |jб|ksЖJВdSrL) r$rMr#r;r:rNr>rr?rTrUrrr┌test_2D_pathway_to_loadsЛs (&■z*TestLinearPathway.test_2D_pathway_to_loadscCs||jа|j|j|jj|j|jjd|j|jj бt |jd|jdd|jdГ}t|jj |ГdksxJВdSйNr&щr)r$rMr#r;r:rNr<┌yr=┌zrrr?rOrUrrr┌test_3D_pathway_lengthУs,■&z(TestLinearPathway.test_3D_pathway_lengthcCs░|jа|j|j|jj|j|jjd|j|jj бt |jd|jdd|jdГ}|j|j||j|j|d|j|j |}t|jj|ГdksмJВdSrZ)r$rMr#r;r:rNr<r\r=r]rr@rArBrr?rR)r!rOrJrrr┌"test_3D_pathway_extension_velocityЫs,■& ■ z4TestLinearPathway.test_3D_pathway_extension_velocitycCs*|jа|j|j|jj|j|jjd|j|jj бt |jd|jdd|jdГ}|j|j|jj||j|j|jj|d|j|j|jj |}|j|j|jj||j|j|jj|d|j|j|jj |}|j|f|j|fg}|jа |jб|kРs&JВdS)Nr&r[)r$rMr#r;r:rNr<r\r=r]rr>r?rT)r!rO┌pO_forceZpI_forcerJrrr┌test_3D_pathway_to_loadsиs*,■& ■ ■ ■z*TestLinearPathway.test_3D_pathway_to_loads)┌__name__┌ __module__┌__qualname__r"┌staticmethodr1┌mark┌parametrizerr/r6r8┌fixturerCrGrKrPrSrVrWrXrYr^r_rarrrrr$sJ ■ ■■ ■■ rc @s~eZdZddДZeejаddddДdDГfdd dДd DГfdddДd DГfdddДdDГfgбddДГГZeejаde dГge dГe dГggбddДГГZ eejаdde dГe dГfe dГde dГfe dГe dГdfgбddДГГZddДZeejаdddДdDГd fd!dДd DГd"fd#dДd DГd$fgбd%d&ДГГZ ejd'd(Нd)d*ДГZd+d,ДZd-d.ДZd/d0ДZd1d2ДZd3d4ДZd5d6ДZdS)7┌TestObstacleSetPathwaycCstttГsJВdSr)rr rr rrrr"┴sz4TestObstacleSetPathway.test_is_pathway_base_subclassznum_attachments, attachmentsщcCsg|]}t|ГСqSrйrйrr+rrrr╚єz!TestObstacleSetPathway.)┌pOr#┌pIr[cCsg|]}t|ГСqSrrkrlrrrr╔rm)rnr#r$roщcCsg|]}t|ГСqSrrkrlrrrr╩rm)rnr#r$┌pCroщcCsg|]}t|ГСqSrrkrlrrrr╦rm)rnr#r$rqZpDrocCsTt|О}t|tГsJВt|dГs$JВt|jГ|ks6JВ|jD]}t|tГs0YtаtбПd|jd<WdГn1sr0YtаtбПd|jd<WdГn1sж0YtаtбПd|jd<WdГn1s┌0YdS)Nrnr#rorr'щ )r rr1r2rEr%r7йr!r?rrrrGьs$((z4TestObstacleSetPathway.test_properties_are_immutablezattachments, expectedcCsg|]}t|ГСqSrrkrlrrrr№rmzObstacleSetPathway(pO, pA, pI)cCsg|]}t|ГСqSrrkrlrrrrrmz"ObstacleSetPathway(pO, pA, pB, pI)cCsg|]}t|ГСqSrrkrlrrrrrmz&ObstacleSetPathway(pO, pA, pB, pC, pI)cCst|О}t|Г|ksJВdSr)r rH)r%rJr?rrrrKўsz TestObstacleSetPathway.test_reprTr9cCsVtdГ|_tdГ|_tdГ|_tdГ|_tdГ|_tdГ|_tddГ|_ t dГ|_dS) Nr:rnror#r$┌qr'r>)r r:rrnror#r$rry┌qdrr>r rrr┌_obstacle_set_pathway_fixture s z4TestObstacleSetPathway._obstacle_set_pathway_fixturecCsp|jа|j|jjб|jа|j|jjб|jа|j|jjбt |j|j|j|jГ}|j ddtdГkslJВdS)Nr'r&)r#rMrnr:rNr$r\ror]r rOrrxrrrrPs z1TestObstacleSetPathway.test_static_pathway_lengthcCsd|jа|j|jjб|jа|j|jjб|jа|j|jjбt |j|j|j|jГ}|j dks`JВdSйNr)r#rMrnr:rNr$r\ror]r rRrxrrrrSs z=TestObstacleSetPathway.test_static_pathway_extension_velocityc Cs:|jа|j|jjб|jа|j|jjб|jа|j|jjбt |j|j|j|jГ}t |j|j|jjГt |j|j|jjГt |j|jtdГd|jj|jjГt |j|jtdГd|jj|jjГt |j|jtdГd|jj|jjГt |j|jtdГd|jj|jjГg}|а |jб|kРs6JВdSrL)r#rMrnr:rNr$r\ror]r rr>rrTrIrrrrV&s((((·z3TestObstacleSetPathway.test_static_pathway_to_loadscCs╪|jа|j|jj|jjб|jа|jt|jГ|jjt |jГd|jjб|j а|jt |jГ|jjt|jГd|jjбt|j|j|j|j Г}dtdГtddt|jГГ}|j |абdks╘JВdSйNr'r&r)r#rMrnr:rNr\r$rryrror rrOrrIrrrrW5s* * "z-TestObstacleSetPathway.test_2D_pathway_lengthcCsц|jа|j|jj|jjб|jа|jt|jГ|jjt |jГd|jjб|j а|jt |jГ|jjt|jГd|jjбt|j|j|j|j Г}tdГt |jГ|j dtt|jГdГ}|j|абdksтJВdSr})r#rMrnr:rNr\r$rryrror rrzrRrrIrrrrXAs* * 0z9TestObstacleSetPathway.test_2D_pathway_extension_velocityc Cs╚|jа|j|jj|jjб|jа|jt|jГ|jjt |jГd|jjб|j а|jt |jГ|jjt|jГd|jjбt|j|j|j|j Г}tdГd|jj|jj}tdt|jГdГd|jjtdГt |jГdtt|jГdГ|jj}t|jt dГ|jjt |jt dГ|jj}t|j|j|Гt|j|j|Гt|j|j|Гt|j|j|Гt|j|j|Гt|j |j|Гg}t|а|jбГ|kРs─JВdS)Nr'r&r[)r#rMrnr:rNr\r$rryrror rrrr>rrT)r!r?ZpO_pA_force_vecZpA_pB_force_vecZpB_pI_force_vecrJrrrrYMs,* * ". 4·z/TestObstacleSetPathway.test_2D_pathway_to_loads)rbrcrdr"rer1rfrgr/rrtrvrGrKrhr{rPrSrVrWrXrYrrrrri┐sd№■ ■¤■■■■ў■ ric@s@eZdZddДZejddНddДГZddДZejа d e d Гfe d Гe dГe dГfgбd dДГZeejа d de dГfe d ГdfgбddДГГZ ddДZejа dedГedГedГedГjgбddДГZddДZddДZddДZed d!ДГZejа d"d#d$ed%fd$ed%Гd%ed%Гd%d&fd'ed(fd#ed)d%Гed'Гd%d&fed'fgбd*d+ДГZejа d"d#d$ed)d%Гefd#d,efd,d#efd$ed%Гd%ed%Гd%d&fd-ed(fd#ed)d%Гed'Гd%d&fed'fd$ed%Гed)d%Гed%Гed)d%Гd)fed)ed-d(Гed%Гfd#ed)d%Гed'Гed)d%Гd)fed)ed)d'Гed%Гfgбd.d/ДГZejа d0d1d$ed%Гed)d%Гed%Гed)d%Гd&ffd#ed)d%Гed'Гed)d%Гd&ffgбd2d3ДГZejа d0d1d4d5d$ed%Гd%ed%Гd%d&ffd#ed)d%Гed'Гd%d&ffd$ed%Гed)d%Гed%Гd%d)ffd#ed)d%Гed'Гd%d)ffgбd6d7ДГZejа d8d9d$ed%Гd%ed%Гd%d&fd#ed%Гd%ed%Гd%d&fd:ed%Гd%ed%Гd%d&ffd#ed)d%Гed'Гd%d&fd$ed'Гd%ed)d%Гd&fed'Гd%ed)d%Гd)d&fffбd;d<ДГZejа d8d9d=d>d$ed%Гd%ed%Гd%d&fd,ed%Гd%ed%Гd%d&fd)ed%Гd%ed%Гd%d&ffd#ed)d%Гed'Гd%d&fd$ed'Гd%ed)d%Гd&fed'Гd%ed)d%Гd)d&ffd#ed%Гd%ed%Гd%d&fd$ed%Гd%ed%Гd%d&fed%Гd%ed%Гd%d)d&ffd?d$ed%Гd%ed%Гd%d)fd@eedAdBed%Гd&d(edAdBed%Гfd@ed%Гed%edAdBed%Гd@ed%Гed%edAdBed%ГdCedAdBed%Гfd-ed%Гd%ed%edAdBed%Гd-ed%Гed%edAdBed%Гd&fffбdDdEДГZ dFdGДZ!dHdIДZ"dJdKДZ#dS)L┌TestWrappingPathwaycCstttГsJВdSr)rrrr rrrr"isz1TestWrappingPathway.test_is_pathway_base_subclassTr9cCsЖtdГ|_tdГ|_tdddН|_tdГ|_tdГ|_|jj|_ t |j|jГ|_t|j|j|j Г|_ t|j|j|j Г|_tdГ|_dS) Nr#r$┌rT)Zpositivernr:r>)rr#r$rrrnr r:r]┌axr┌spherer┌cylinderrr?r>r rrr┌_wrapping_pathway_fixturels z-TestWrappingPathway._wrapping_pathway_fixturecCs┬t|j|j|jГ}t|tГs JВt|dГs.JВt|jГdks@JВt|jdtГsTJВ|jd|jkshJВt|jdtГs|JВ|jd|jksРJВt|dГsЮJВt|j t ГsоJВ|j |jks╛JВdS)Nr%r&rr'┌geometry)rr#r$rВr(r)r*r%rrДrrFrrrr/ysz*TestWrappingPathway.test_valid_constructorr%r#r$r0cCsBtаtбП$tg|в|jСRО}WdГn1s40YdSr)r1r2r7rrВ)r!r%r5rrrrtЖszITestWrappingPathway.test_invalid_constructor_attachments_incorrect_numberNcCs6tаtбПt|О}WdГn1s(0YdSrrur4rrrrvСs zBTestWrappingPathway.test_invalid_constructor_attachments_not_pointcCs<tаtбПt|j|jГ}WdГn1s.0YdSrйr1r2r7rr#r$)r!r5rrr┌1test_invalid_constructor_geometry_is_not_suppliedЭszETestWrappingPathway.test_invalid_constructor_geometry_is_not_suppliedrДrryr:cCs>tаtбП t|j|j|Г}WdГn1s00YdSrrЕ)r!rДr5rrr┌"test_invalid_geometry_not_geometryбs z6TestWrappingPathway.test_invalid_geometry_not_geometrycCsttаtбП|j|jjd<WdГn1s.0YtаtбП|j|jjd<WdГn1sf0YdSrD)r1r2r7r$r?r%r#r rrr┌&test_attachments_property_is_immutableоs,z:TestWrappingPathway.test_attachments_property_is_immutablecCs6tаtбПd|j_WdГn1s(0YdSr)r1r2rEr?rДr rrr┌#test_geometry_property_is_immutable┤sz7TestWrappingPathway.test_geometry_property_is_immutablecCs$d|jЫdЭ}t|jГ|ks JВdS)Nz!WrappingPathway(pA, pB, geometry=·))rВrHr?rUrrrrK╕s zTestWrappingPathway.test_reprcCstddДt||ГDГГS)Ncss|]\}}||VqdSrrйrZmag┌unitrrr┌ ┴rmz9TestWrappingPathway._expand_pos_to_vec..)┌sum┌zip)┌pos┌framerrr┌_expand_pos_to_vec┐sz&TestWrappingPathway._expand_pos_to_veczpA_vec, pB_vec, factorйr'rrйrr'rr&rrjr[r'cCs~|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}||j}t |j |ГdkszJВdSr|)rТr:r#rMrnrr$rrБrrOйr!┌pA_vec┌pB_vecZfactorr?rJrrr┌$test_static_pathway_on_sphere_length├s z8TestWrappingPathway.test_static_pathway_on_sphere_lengthйrwrrrpcCsЖ|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|t |jdГ}t |j|ГdksВJВdSrQ)rТr:r#rMrnrr$rrВrrrOrХrrr┌&test_static_pathway_on_cylinder_length╘sz:TestWrappingPathway.test_static_pathway_on_cylinder_lengthzpA_vec, pB_vec)rУrФcCsl|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|j dkshJВdSr|) rТr:r#rMrnrr$rrБrRйr!rЦrЧr?rrr┌0test_static_pathway_on_sphere_extension_velocityёs zDTestWrappingPathway.test_static_pathway_on_sphere_extension_velocity)rУrЩ)rЩrУcCsl|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|j dkshJВdSr|) rТr:r#rMrnrr$rrВrRrЫrrr┌2test_static_pathway_on_cylinder_extension_velocitys zFTestWrappingPathway.test_static_pathway_on_cylinder_extension_velocityzApA_vec, pB_vec, pA_vec_expected, pB_vec_expected, pO_vec_expected)rУrФrФrУ)rwrwrrwc Cs>|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}t ddДt ||jГDГГ}t ddДt ||jГDГГ}t ddДt ||jГDГГ}t|j|j|jdt |jdГ|Гt|j|j|jdt |jdГ|Гt|j|j|jdt |jdГ|Гg}|а|jб|kРs:JВdS)Ncss|]\}}||VqdSrrrЛrrrrН7szMTestWrappingPathway.test_static_pathway_on_sphere_to_loads..css|]\}}||VqdSrrrЛrrrrН:scss|]\}}||VqdSrrrЛrrrrН=srjrr)rТr:r#rMrnrr$rrБrОrПrr>rrT)r!rЦrЧ┌pA_vec_expected┌pB_vec_expected┌pO_vec_expectedr?rJrrr┌&test_static_pathway_on_sphere_to_loadss& (((¤z:TestWrappingPathway.test_static_pathway_on_sphere_to_loads)rУrЩrФrФ)rщ■ r)rЩrУйrrwrrг)rr&r)rФ)rr'r')rrr')rrrw)rrrщ√ щщщ№ cCs╘|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|j |а||jб}|j |а||jб}|j |а||jб} t |j|Гt |j|Гt |j| Гg} t|а|j бГ| ks╨JВdSr) rТr:r#rMrnrr$rrВr>rrrT)r!rЦrЧrЮrЯrаr?ZpA_force_expectedZpB_force_expectedZpO_force_expectedrJrrr┌(test_static_pathway_on_cylinder_to_loadsGs&5 ¤zrrr?rT) r!ryrйrкZpA_forceZpB_forcer`rJrrrr┌$test_2D_pathway_on_cylinder_to_loadsвs&&, ¤z8TestWrappingPathway.test_2D_pathway_on_cylinder_to_loads)$rbrcrdr"r1rhrГr/rfrgrrtrervrЖrrr rNrЗrИrЙrKrТrrrrШrЪrЬrЭrбrиrлrмrнrrrrr~gs ■■ ■■№■ ( ¤■ ( &¤¤ї■ * ¤■ $∙■ √√ў■ √√√4&&¤*&¤ўц■- r~)┌__doc__r1ZsympyrrrrrrZsympy.physics.mechanicsrr r rrr rrrrrZsympy.simplify.simplifyrrrrir~rrrr┌s 4 )