U Z╩╣jpaу@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ГstВdSйN)┌ issubclassr r┌AssertionErrorй┌selfrrr┌test_is_pathway_base_subclass&sz/TestLinearPathway.test_is_pathway_base_subclasszargs, kwargs┌pA┌pBcCs╕|\}}t||О}t|tГs tВt|dГs.tВt|jГdks@tВ|jd|ksRtВ|jd|ksdtВt|jdtГsxtВ|jdjdksМtВt|jdtГsаtВ|jdjdks┤tВdS)N┌attachmentsщrщr$r%)r ┌ isinstancer ┌hasattr┌lenr&r┌name)┌args┌kwargsZpointAZpointB┌instancerrr┌test_valid_constructor)s z(TestLinearPathway.test_valid_constructorr&┌pZc Cs"tаtбПt|О}W5QRXdSr)┌pytest┌raises┌ ValueErrorr йr&┌_rrr┌)test_invalid_attachments_incorrect_number=s z;TestLinearPathway.test_invalid_attachments_incorrect_numberNc Cs"tаtбПt|О}W5QRXdSr)r2r3┌ TypeErrorr r5rrr┌"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_fixturec Csnt|j|jГ}tаtбПd|_W5QRXtаtбПd|jd<W5QRXtаtбПd|jd<W5QRXdSйNrr()r r$r%r2r3┌AttributeErrorr&r8йr"r/rrr┌test_properties_are_immutablecsz/TestLinearPathway.test_properties_are_immutablecCs&t|j|jГ}d}t|Г|ks"tВdS)NzLinearPathway(pA, pB))r r$r%┌reprr йr"r@┌expectedrrr┌ test_reprlszTestLinearPathway.test_reprcCs,|jа|jd|jjб|jjdks(tВdSйNr')r%┌set_posr$r;┌xr@┌lengthr r!rrr┌test_static_pathway_lengthqsz,TestLinearPathway.test_static_pathway_lengthcCs,|jа|jd|jjб|jjdks(tВdSйNr'r)r%rNr$r;rOr@┌extension_velocityr r!rrr┌&test_static_pathway_extension_velocityusz8TestLinearPathway.test_static_pathway_extension_velocitycCs\|jа|jd|jjб|j|j|jjf|j|j|jjfg}|jа|jб|ksXtВdSrM) r%rNr$r;rOr?r@┌to_loadsr йr"rKrrr┌test_static_pathway_to_loadsys ■z.TestLinearPathway.test_static_pathway_to_loadscCsD|jа|jd|j|jjбdt|jdГ}|jj|ks@t ВdSrM) r%rNr$r<r;rOrr@rPr rVrrr┌test_2D_pathway_lengthБsz(TestLinearPathway.test_2D_pathway_lengthcCsP|jа|jd|j|jjбdt|jdГ|j|j}|jj |ksLt ВdSrM)r%rNr$r<r;rOrrAr@rSr rVrrr┌"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Жt ВdSrM)r%rNr$r<r;rOr?rr@rUr rVrrr┌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 |ГdksxtВdSйNr'щr)r%rNr$r<r;rOr=┌yr>┌zrrr@rPr rVrrr┌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мtВdSr[)r%rNr$r<r;rOr=r]r>r^rrArBrCrr@rSr )r"rPrKrrr┌"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&tВdS)Nr'r\)r%rNr$r<r;rOr=r]r>r^rr?r@rUr )r"rP┌pO_forceZpI_forcerKrrr┌test_3D_pathway_to_loadsиs*,■& ■ ■ ■z*TestLinearPathway.test_3D_pathway_to_loads)┌__name__┌ __module__┌__qualname__r#┌staticmethodr2┌mark┌parametrizerr0r7r9┌fixturerDrHrLrQrTrWrXrYrZr_r`rbrrrrr$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ГstВdSr)rr rr r!rrrr#┴sz4TestObstacleSetPathway.test_is_pathway_base_subclassznum_attachments, attachmentsщcCsg|]}t|ГСqSrйrйrr,rrrr╚sz!TestObstacleSetPathway.)┌pOr$┌pIr\cCsg|]}t|ГСqSrrlrmrrrr╔s)rnr$r%roщcCsg|]}t|ГСqSrrlrmrrrr╩s)rnr$r%┌pCroщcCsg|]}t|ГСqSrrlrmrrrr╦s)rnr$r%rqZpDrocCsTt|О}t|tГstВt|dГs$tВt|jГ|ks6tВ|jD]}t|tГsd$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ГstВdSr)rrrr r!rrrr#isz1TestWrappingPathway.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)Zpositivernr;r?)rr$r%rrrnr r;r^Zaxr┌spherer┌cylinderrr@r?r!rrr┌_wrapping_pathway_fixturels z-TestWrappingPathway._wrapping_pathway_fixturecCs┬t|j|j|jГ}t|tГs tВt|dГs.tВt|jГdks@tВt|jdt ГsTtВ|jd|jkshtВt|jdt Гs|tВ|jd|jksРtВt|dГsЮtВt|j tГsоtВ|j |jks╛tВdS)Nr&r'rr(┌geometry)rr$r%rБr)r r*r+r&rrГrrGrrrr0ysz*TestWrappingPathway.test_valid_constructorr&r$r%r1c Cs*tаtбПt||jfЮО}W5QRXdSr)r2r3r8rrБ)r"r&r6rrrrtЖszITestWrappingPathway.test_invalid_constructor_attachments_incorrect_numberNc Cs"tаtбПt|О}W5QRXdSrrur5rrrrvСs zBTestWrappingPathway.test_invalid_constructor_attachments_not_pointc Cs(tаtбПt|j|jГ}W5QRXdSrйr2r3r8rr$r%)r"r6rrr┌1test_invalid_constructor_geometry_is_not_suppliedЭszETestWrappingPathway.test_invalid_constructor_geometry_is_not_suppliedrГrryr;c Cs*tаtбПt|j|j|Г}W5QRXdSrrД)r"rГr6rrr┌"test_invalid_geometry_not_geometryбs z6TestWrappingPathway.test_invalid_geometry_not_geometryc CsLtаtбП|j|jjd<W5QRXtаtбП|j|jjd<W5QRXdSrE)r2r3r8r%r@r&r$r!rrr┌&test_attachments_property_is_immutableоsz:TestWrappingPathway.test_attachments_property_is_immutablec Cs"tаtбПd|j_W5QRXdSr)r2r3rFr@rГr!rrr┌#test_geometry_property_is_immutable┤sz7TestWrappingPathway.test_geometry_property_is_immutablecCs$d|jЫdЭ}t|jГ|ks tВdS)Nz!WrappingPathway(pA, pB, geometry=·))rБrIr@r rVrrrrL╕s zTestWrappingPathway.test_reprcCstddДt||ГDГГS)Ncss|]\}}||VqdSrrйr┌mag┌unitrrr┌ ┴sz9TestWrappingPathway._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'rrkr\r(cCs~|а||jб}|а||jб}|jа|j|j|б|jа|j|j|бt|j|j|jГ}||j}t |j |ГdksztВdSr|)rТr;r$rNrnrr%rrАrrPr йr"┌pA_vec┌pB_vec┌factorr@rKrrr┌$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ВtВdSrR) rТr;r$rNrnrr%rrБrrrPr rХ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 dksht ВdSr|)rТr;r$rNrnrr%rrАrSr й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 dksht ВdSr|)rТr;r$rNrnrr%rrБrSr 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У)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:tВdS)Ncss|]\}}||VqdSrrrКrrrrН7szMTestWrappingPathway.test_static_pathway_on_sphere_to_loads..css|]\}}||VqdSrrrКrrrrН:scss|]\}}||VqdSrrrКrrrrН=srkrr)rТr;r$rNrnrr%rrАrОrПrr?rrUr )r"rЦrЧ┌pA_vec_expected┌pB_vec_expected┌pO_vec_expectedr@rKrrr┌&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╨t ВdSr)rТr;r$rNrnrr%rrБr?rrrUr )r"rЦrЧrЯrаrбr@ZpA_force_expectedZpB_force_expectedZpO_force_expectedrKrrr┌(test_static_pathway_on_cylinder_to_loadsGs&5 ¤zs 4 )