o Е║еgpaу@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┌kwargs┌pointA┌pointB┌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_numberNcCr6r )r7r8┌ TypeErrorr r:rrr┌"test_invalid_attachments_not_pointIr=z4TestLinearPathway.test_invalid_attachments_not_pointTй┌autousecCs~tdГ|_tdГ|_tdГ|_t|j|jГ|_tdГ|_tdГ|_ tdГ|_ tddГ|_tddГ|_tddГ|_ tdГ|_dS) N┌Nr&r'┌q1┌q2┌q3r*┌F)r rBrr&r'r ┌pathwayrrCrDrE┌q1d┌q2d┌q3drrFr"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'r7r8┌AttributeErrorr(r>йr#r3rrr┌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#rG┌expectedrrr┌ test_reprlszTestLinearPathway.test_reprcCs,|jа|jd|jjб|jjdksJВdSйNr))r'┌set_posr&rB┌xrG┌lengthr"rrr┌test_static_pathway_lengthqєz,TestLinearPathway.test_static_pathway_lengthcCs,|jа|jd|jjб|jjdksJВdSйNr)r)r'rUr&rBrVrG┌extension_velocityr"rrr┌&test_static_pathway_extension_velocityurYz8TestLinearPathway.test_static_pathway_extension_velocitycCs\|jа|jd|jjб|j|j|jjf|j|j|jjfg}|jа|jб|ks,JВdSrT)r'rUr&rBrVrFrG┌to_loadsйr#rRrrr┌test_static_pathway_to_loadsys ■z.TestLinearPathway.test_static_pathway_to_loadscCsD|jа|jd|j|jjбdt|jdГ}|jj|ks JВdSrT) r'rUr&rCrBrVrrGrWr^rrr┌test_2D_pathway_lengthБsz(TestLinearPathway.test_2D_pathway_lengthcCsP|jа|jd|j|jjбdt|jdГ|j|j}|jj |ks&JВdSrT) r'rUr&rCrBrVrrHrGr[r^rrr┌"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ВdSrT) r'rUr&rCrBrVrFrrGr]r^rrr┌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 |Гdks)┌pOr&┌pIrdcCrurrvrwrrrr╔rx)ryr&r'rzщcCrurrvrwrrrr╩rx)ryr&r'┌pCrzщcCrurrvrwrrrr╦rx)ryr&r'r|┌pDrzcCsTt|О}t|tГsJВt|dГsJВt|jГ|ksJВ|jD] }t|tГs'JВqdS)Nr()r r+r,r-r(r)┌num_attachmentsr(r3┌ attachmentrrrr4─s z-TestObstacleSetPathway.test_valid_constructorr(ryrzcCr6r )r7r8r9r r:rrr┌5test_invalid_constructor_attachments_incorrect_number╓s " zLTestObstacleSetPathway.test_invalid_constructor_attachments_incorrect_numberNr&cCr6r йr7r8r>rr:rrr┌.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)Nryr&rzrr*щ )r rr7r8rMr(r>йr#rGrrrrOьs " z4TestObstacleSetPathway.test_properties_are_immutablezattachments, expectedcCrurrvrwrrrr№rxzObstacleSetPathway(pO, pA, pI)cCrurrvrwrrrrrxz"ObstacleSetPathway(pO, pA, pB, pI)cCrurrvrwrrrrrxz&ObstacleSetPathway(pO, pA, pB, pC, pI)cCst|О}t|Г|ksJВdSr )r rP)r(rRrGrrrrSўsz TestObstacleSetPathway.test_reprTr@cCsVtdГ|_tdГ|_tdГ|_tdГ|_tdГ|_tdГ|_tddГ|_ t dГ|_dS) NrBryrzr&r'┌qr*rF)r rBrryrzr&r'rrЖ┌qdrrFr"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&rUryrBrVr'rerzrfr rWrrЕrrrrXs z1TestObstacleSetPathway.test_static_pathway_lengthcCsd|jа|j|jjб|jа|j|jjб|jа|j|jjбt |j|j|j|jГ}|j dks0JВdSйNr)r&rUryrBrVr'rerzrfr r[rЕrrrr\s 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ВdSrT)r&rUryrBrVr'rerzrfr rrFrr]rQrrrr_&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&rUryrBrVrer'rrЖrrzr rrWrrQrrrr`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&rUryrBrVrer'rrЖrrzr rrЗr[rrQrrrraAs* * 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)rd)r&rUryrBrVrer'rrЖrrzr rrrrFrr])r#rG┌pO_pA_force_vec┌pA_pB_force_vec┌pB_pI_force_vecrRrrrrbMs,* * ". 4·z/TestObstacleSetPathway.test_2D_pathway_to_loads)rlrmrnr$ror7rprqr4rrБrГrOrSrrrИrXr\r_r`rarbrrrrrs┐sf№■ ■¤■■■■ў■ rsc@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)┌positiveryrBrF)rr&r'rrПryr rBrf┌axr┌spherer┌cylinderrrGrFr"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ХrrNrrrr4ysz*TestWrappingPathway.test_valid_constructorr(r&r'r5cCsDtаtбПtg|в|jСRО}WdГdS1swYdSr )r7r8r>rrУ)r#r(r;rrrrБЖs" zITestWrappingPathway.test_invalid_constructor_attachments_incorrect_numberNcCr6r rВr:rrrrГСr=zBTestWrappingPathway.test_invalid_constructor_attachments_not_pointcCs>tаtбПt|j|jГ}WdГdS1swYdSr йr7r8r>rr&r')r#r;rrr┌1test_invalid_constructor_geometry_is_not_suppliedЭs" zETestWrappingPathway.test_invalid_constructor_geometry_is_not_suppliedrХrПrЖrBcCs@tаtбПt|j|j|Г}WdГdS1swYdSr rЦ)r#rХr;rrr┌"test_invalid_geometry_not_geometryбs " z6TestWrappingPathway.test_invalid_geometry_not_geometrycCsvtаtбП|j|jjd<WdГn1swYtаtбП|j|jjd<WdГdS1s4wYdSrL)r7r8r>r'rGr(r&r"rrr┌&test_attachments_property_is_immutableоs " z:TestWrappingPathway.test_attachments_property_is_immutablecCs8tаtбП d|j_WdГdS1swYdSr )r7r8rMrGrХ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УrPrGr^rrrrS╕s zTestWrappingPathway.test_reprcCstddДt||ГDГГS)NcsєБ|] \}}||VqdSr rй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)rrtrdr*cCs~|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}||j}t |j |Гdks=JВdSrЙ)rеrBr&rUryrПr'rrТrrWйr#┌pA_vec┌pB_vec┌factorrGrRrrr┌$test_static_pathway_on_sphere_length├s z8TestWrappingPathway.test_static_pathway_on_sphere_lengthйrДrrr{cCsЖ|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}|t |jdГ}t |j|ГdksAJВdSrZ)rеrBr&rUryrПr'rrУrrrWrи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еrBr&rUryrПr'rrТr[йr#rйrкrGrrr┌0test_static_pathway_on_sphere_extension_velocityёs zDTestWrappingPathway.test_static_pathway_on_sphere_extension_velocity)rжrн)rнrжcCrпrЙ) rеrBr&rUryrПr'rrУr[r░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Ьr rrЭrrrrа7єА zMTestWrappingPathway.test_static_pathway_on_sphere_to_loads..csrЬr rrЭrrrrа:r│csrЬr rrЭrrrrа=r│rtr})rеrBr&rUryrПr'rrТrбrвrrFrr])r#rйrк┌pA_vec_expected┌pB_vec_expected┌pO_vec_expectedrGrRrrr┌&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еrBr&rUryrПr'rrУrFrrr])r#rйrкr┤r╡r╢rG┌pA_force_expected┌pB_force_expected┌pO_force_expectedrRrrr┌(test_static_pathway_on_cylinder_to_loadsGs&5 ¤zs 4 )