o GZО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))┌TestLinearPathwaycCєtttГsJВdSйN)┌ issubclassr rй┌selfrrr┌test_is_pathway_base_subclass&єz/TestLinearPathway.test_is_pathway_base_subclasszargs, kwargs┌pA┌pBcCs╝|\}}t|i|дО}t|tГsJВt|dГsJВt|jГdks"JВ|jd|us+JВ|jd|us4JВt|jdtГs>JВ|jdjdksHJВt|jdtГsRJВ|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'┌pZcCє8tаtбП t|О}WdГdS1swYdSr)┌pytest┌raises┌ ValueErrorr йr'┌_rrr┌)test_invalid_attachments_incorrect_number=є " z;TestLinearPathway.test_invalid_attachments_incorrect_numberNcCr3r)r4r5┌ TypeErrorr r7rrr┌"test_invalid_attachments_not_pointIr: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@rA┌q1d┌q2d┌q3drrBr!rrr┌_linear_pathway_fixtureUs z)TestLinearPathway._linear_pathway_fixturecCsмt|j|jГ}tаtбПd|_WdГn1swYtаtбП d|jd<WdГn1s4wYtаtбПd|jd<WdГdS1sOwYdSйNrr))r r%r&r4r5┌AttributeErrorr'r;йr"r0rrr┌test_properties_are_immutablecs " z/TestLinearPathway.test_properties_are_immutablecCs&t|j|jГ}d}t|Г|ksJВdS)NzLinearPathway(pA, pB))r r%r&┌reprйr"rC┌expectedrrr┌ test_reprlszTestLinearPathway.test_reprcCs,|jа|jd|jjб|jjdksJВdSйNr()r&┌set_posr%r>┌xrC┌lengthr!rrr┌test_static_pathway_lengthqєz,TestLinearPathway.test_static_pathway_lengthcCs,|jа|jd|jjб|jjdksJВdSйNr(r)r&rQr%r>rRrC┌extension_velocityr!rrr┌&test_static_pathway_extension_velocityurUz8TestLinearPathway.test_static_pathway_extension_velocitycCs\|jа|jd|jjб|j|j|jjf|j|j|jjfg}|jа|jб|ks,JВdSrP)r&rQr%r>rRrBrC┌to_loadsйr"rNrrr┌test_static_pathway_to_loadsys ■z.TestLinearPathway.test_static_pathway_to_loadscCsD|jа|jd|j|jjбdt|jdГ}|jj|ks JВdSrP) r&rQr%r?r>rRrrCrSrZrrr┌test_2D_pathway_lengthБsz(TestLinearPathway.test_2D_pathway_lengthcCsP|jа|jd|j|jjбdt|jdГ|j|j}|jj |ks&JВdSrP) r&rQr%r?r>rRrrDrCrWrZrrr┌"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б|ksCJВdSrP) r&rQr%r?r>rRrBrrCrYrZrrr┌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 |ГdksrRr@┌yrA┌zrrrCrSrZrrr┌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|ГdksVJВdSr_)r&rQr%r?r>rRr@rarArbrrDrErFrrCrW)r"rSrNrrr┌"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б|ksТJВdS)Nr(r`)r&rQr%r?r>rRr@rarArbrrBrCrY)r"rS┌pO_forceZpI_forcerNrrr┌test_3D_pathway_to_loadsиs*,■& ■ ■ ■z*TestLinearPathway.test_3D_pathway_to_loads)┌__name__┌ __module__┌__qualname__r#┌staticmethodr4┌mark┌parametrizerr1r9r<┌fixturerGrKrOrTrXr[r\r]r^rcrdrfrrrrr$sL ■ ■■ ■■ 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┌TestObstacleSetPathwaycCrr)r r rr!rrrr#┴r$z4TestObstacleSetPathway.test_is_pathway_base_subclassznum_attachments, attachmentsщcCєg|]}t|ГСqSrйrйrr-rrrr╚єz!TestObstacleSetPathway.)┌pOr%┌pIr`cCrprrqrrrrrr╔rs)rtr%r&ruщcCrprrqrrrrrr╩rs)rtr%r&┌pCruщcCrprrqrrrrrr╦rs)rtr%r&rwZpDrucCsTt|О}t|tГsJВt|dГsJВt|jГ|ksJВ|jD] }t|tГs'JВqdS)Nr')r r*r+r,r'r)Znum_attachmentsr'r0┌ attachmentrrrr1─s z-TestObstacleSetPathway.test_valid_constructorr'rtrucCr3r)r4r5r6r r7rrr┌5test_invalid_constructor_attachments_incorrect_number╓s " zLTestObstacleSetPathway.test_invalid_constructor_attachments_incorrect_numberNr%cCr3rйr4r5r;rr7rrr┌.test_invalid_constructor_attachments_not_point▀s " zETestObstacleSetPathway.test_invalid_constructor_attachments_not_pointcCsъttdГtdГtdГГ}tаtбПd|_WdГn1swYtаtбП d|jd<WdГn1s9wYtаtбП d|jd<WdГn1sSwYtаtбПd|jd<WdГdS1snwYdS)Nrtr%rurr)щ )r rr4r5rIr'r;йr"rCrrrrKьs " z4TestObstacleSetPathway.test_properties_are_immutablezattachments, expectedcCrprrqrrrrrr№rszObstacleSetPathway(pO, pA, pI)cCrprrqrrrrrrrsz"ObstacleSetPathway(pO, pA, pB, pI)cCrprrqrrrrrrrsz&ObstacleSetPathway(pO, pA, pB, pC, pI)cCst|О}t|Г|ksJВdSr)r rL)r'rNrCrrrrOўsz TestObstacleSetPathway.test_reprTr=cCsVtdГ|_tdГ|_tdГ|_tdГ|_tdГ|_tdГ|_tddГ|_ t dГ|_dS) Nr>rtrur%r&┌qr)rB)r r>rrtrur%r&rr┌qdrrBr!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Гks6JВdS)Nr)r()r%rQrtr>rRr&rarurbr rSrr~rrrrTs z1TestObstacleSetPathway.test_static_pathway_lengthcCsd|jа|j|jjб|jа|j|jjб|jа|j|jjбt |j|j|j|jГ}|j dks0JВdSйNr)r%rQrtr>rRr&rarurbr rWr~rrrrXs z=TestObstacleSetPathway.test_static_pathway_extension_velocityc Cs8|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б|ksЪJВdSrP)r%rQrtr>rRr&rarurbr rrBrrYrMrrrr[&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 |абdksjJВdSйNr)r(r)r%rQrtr>rRrar&rrrrur rrSrrMrrrr\5s* * "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|абdksqJВdSrГ)r%rQrtr>rRrar&rrrrur rrАrWrrMrrrr]As* * 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бГ|ksсJВdS)Nr)r(r`)r%rQrtr>rRrar&rrrrur rrrrBrrY)r"rCZpO_pA_force_vecZpA_pB_force_vecZpB_pI_force_vecrNrrrr^Ms,* * ". 4·z/TestObstacleSetPathway.test_2D_pathway_to_loads)rgrhrir#rjr4rkrlr1rrzr|rKrOrmrБrTrXr[r\r]r^rrrrrn┐sf№■ ■¤■■■■ў■ rnc@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┌TestWrappingPathwaycCrr)r rrr!rrrr#ir$z1TestWrappingPathway.test_is_pathway_base_subclassTr=cCsЖtdГ|_tdГ|_tdddН|_tdГ|_tdГ|_|jj|_ t |j|jГ|_t|j|j|j Г|_ t|j|j|j Г|_tdГ|_dS) Nr%r&┌rT)Zpositivertr>rB)rr%r&rrЕrtr r>rb┌axr┌spherer┌cylinderrrCrBr!rrr┌_wrapping_pathway_fixturels z-TestWrappingPathway._wrapping_pathway_fixturecCs┬t|j|j|jГ}t|tГsJВt|dГsJВt|jГdks JВt|jdtГs*JВ|jd|jks4JВt|jdtГs>JВ|jd|jksHJВt|dГsOJВt|j t ГsWJВ|j |jks_JВdS)Nr'r(rr)┌geometry)rr%r&rИr*r+r,r'rrКrrJrrrr1ysz*TestWrappingPathway.test_valid_constructorr'r%r&r2cCsDtаtбПtg|в|jСRО}WdГdS1swYdSr)r4r5r;rrИ)r"r'r8rrrrzЖs" zITestWrappingPathway.test_invalid_constructor_attachments_incorrect_numberNcCr3rr{r7rrrr|Сr:zBTestWrappingPathway.test_invalid_constructor_attachments_not_pointcCs>tаtбПt|j|jГ}WdГdS1swYdSrйr4r5r;rr%r&)r"r8rrr┌1test_invalid_constructor_geometry_is_not_suppliedЭs" zETestWrappingPathway.test_invalid_constructor_geometry_is_not_suppliedrКrЕrr>cCs@tаtбПt|j|j|Г}WdГdS1swYdSrrЛ)r"rКr8rrr┌"test_invalid_geometry_not_geometryбs " z6TestWrappingPathway.test_invalid_geometry_not_geometrycCsvtаtбП|j|jjd<WdГn1swYtаtбП|j|jjd<WdГdS1s4wYdSrH)r4r5r;r&rCr'r%r!rrr┌&test_attachments_property_is_immutableоs " z:TestWrappingPathway.test_attachments_property_is_immutablecCs8tаtбП d|j_WdГdS1swYdSr)r4r5rIrCrКr!rrr┌#test_geometry_property_is_immutable┤s " z7TestWrappingPathway.test_geometry_property_is_immutablecCs$d|jЫdЭ}t|jГ|ksJВdS)Nz!WrappingPathway(pA, pB, geometry=·))rИrLrCrZrrrrO╕s zTestWrappingPathway.test_reprcCstddДt||ГDГГS)NcsєБ|] \}}||VqdSrrйr┌mag┌unitrrr┌ ┴sАz9TestWrappingPathway._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(rror`r)cCs~|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}||j}t |j |Гdks=JВdSrВ)rЪr>r%rQrtrЕr&rrЗrrSйr"┌pA_vec┌pB_vec┌factorrCrNrrr┌$test_static_pathway_on_sphere_length├s z8TestWrappingPathway.test_static_pathway_on_sphere_lengthйr}rrrvcCsЖ|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|t |jdГ}t |j|ГdksAJВdSrV)rЪr>r%rQrtrЕr&rrИrrrSrЭrrr┌&test_static_pathway_on_cylinder_length╘sz:TestWrappingPathway.test_static_pathway_on_cylinder_lengthzpA_vec, pB_vec)rЫrЬcCєl|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|j dks4JВdSrВ) rЪr>r%rQrtrЕr&rrЗrWйr"rЮrЯrCrrr┌0test_static_pathway_on_sphere_extension_velocityёs zDTestWrappingPathway.test_static_pathway_on_sphere_extension_velocity)rЫrв)rвrЫcCrдrВ) rЪr>r%rQrtrЕr&rrИrWrе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Ы)r}r}rr}c 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б|ksЬJВdS)NcsrСrrrТrrrrХ7єА zMTestWrappingPathway.test_static_pathway_on_sphere_to_loads..csrСrrrТrrrrХ:rиcsrСrrrТrrrrХ=rиrorx)rЪr>r%rQrtrЕr&rrЗrЦrЧrrBrrY)r"rЮrЯ┌pA_vec_expected┌pB_vec_expected┌pO_vec_expectedrCrNrrr┌&test_static_pathway_on_sphere_to_loadss& (((¤z:TestWrappingPathway.test_static_pathway_on_sphere_to_loads)rЫrвrЬrЬ)rщ■ r)rвrЫйrr}rrо)rr(r)rЬ)rr)r))rrr))rrr})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 бГ| kshJВdSr) rЪr>r%rQrtrЕr&rrИrBrrrY)r"rЮrЯrйrкrлrCZpA_force_expectedZpB_force_expectedZpO_force_expectedrNrrr┌(test_static_pathway_on_cylinder_to_loadsGs&5 ¤zrRrrrar%rQrtr&rrrCrS)r"r┌pA_pos┌pB_posrNrrr┌"test_2D_pathway_on_cylinder_lengthПs&z6TestWrappingPathway.test_2D_pathway_on_cylinder_lengthcCsЬtdГ}tddГ}|j|jj}|jt|Г|jjt|Г|jj}|jа|j |б|j а|j |б|jt|dГ||}t|j j|ГdksLJВdS)Nrr)r(r)rrЕr>rRrrrar%rQrtr&rrrCrW)r"rrАr┤r╡rNrrr┌.test_2D_pathway_on_cylinder_extension_velocityШs &zBTestWrappingPathway.test_2D_pathway_on_cylinder_extension_velocityc CstdГ}|j|jj}|jt|Г|jjt|Г|jj}|jа|j |б|j а|j |б|j|jj}|jt|Г|jjt|Г|jj}|jt|Г|jjt|Гd|jj}t|j|Гt|j |Гt|j |Гg}t |jа|jбГ}||ks~JВdS)Nrr))rrЕr>rRrrrar%rQrtr&rBrrrCrY) r"rr┤r╡ZpA_forceZpB_forcererNrrrr┌$test_2D_pathway_on_cylinder_to_loadsвs&&, ¤z8TestWrappingPathway.test_2D_pathway_on_cylinder_to_loads)$rgrhrir#r4rmrЙr1rkrlrrzrjr|rМrrr rRrНrОrПrOrЪrrrrбrгrжrзrмr│r╢r╖r╕rrrrrДgs ■■ ■■№■ ( ¤■ ( &¤¤ї■ * ¤■ $∙■ √√ў■ √√√4&&¤*&¤ўц■ - rД)┌__doc__r4ZsympyrrrrrrZsympy.physics.mechanicsrr r rrr rrrrrZsympy.simplify.simplifyrrrrnrДrrrr┌s 4 )