test_args.py 174 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363
  1. """Test whether all elements of cls.args are instances of Basic. """
  2. # NOTE: keep tests sorted by (module, class name) key. If a class can't
  3. # be instantiated, add it here anyway with @SKIP("abstract class) (see
  4. # e.g. Function).
  5. import os
  6. import re
  7. from sympy.assumptions.ask import Q
  8. from sympy.core.basic import Basic
  9. from sympy.core.function import (Function, Lambda)
  10. from sympy.core.numbers import (Rational, oo, pi)
  11. from sympy.core.relational import Eq
  12. from sympy.core.singleton import S
  13. from sympy.core.symbol import symbols
  14. from sympy.functions.elementary.exponential import (exp, log)
  15. from sympy.functions.elementary.miscellaneous import sqrt
  16. from sympy.functions.elementary.trigonometric import sin
  17. from sympy.testing.pytest import SKIP
  18. a, b, c, x, y, z = symbols('a,b,c,x,y,z')
  19. whitelist = [
  20. "sympy.assumptions.predicates", # tested by test_predicates()
  21. "sympy.assumptions.relation.equality", # tested by test_predicates()
  22. ]
  23. def test_all_classes_are_tested():
  24. this = os.path.split(__file__)[0]
  25. path = os.path.join(this, os.pardir, os.pardir)
  26. sympy_path = os.path.abspath(path)
  27. prefix = os.path.split(sympy_path)[0] + os.sep
  28. re_cls = re.compile(r"^class ([A-Za-z][A-Za-z0-9_]*)\s*\(", re.MULTILINE)
  29. modules = {}
  30. for root, dirs, files in os.walk(sympy_path):
  31. module = root.replace(prefix, "").replace(os.sep, ".")
  32. for file in files:
  33. if file.startswith(("_", "test_", "bench_")):
  34. continue
  35. if not file.endswith(".py"):
  36. continue
  37. with open(os.path.join(root, file), encoding='utf-8') as f:
  38. text = f.read()
  39. submodule = module + '.' + file[:-3]
  40. if any(submodule.startswith(wpath) for wpath in whitelist):
  41. continue
  42. names = re_cls.findall(text)
  43. if not names:
  44. continue
  45. try:
  46. mod = __import__(submodule, fromlist=names)
  47. except ImportError:
  48. continue
  49. def is_Basic(name):
  50. cls = getattr(mod, name)
  51. if hasattr(cls, '_sympy_deprecated_func'):
  52. cls = cls._sympy_deprecated_func
  53. if not isinstance(cls, type):
  54. # check instance of singleton class with same name
  55. cls = type(cls)
  56. return issubclass(cls, Basic)
  57. names = list(filter(is_Basic, names))
  58. if names:
  59. modules[submodule] = names
  60. ns = globals()
  61. failed = []
  62. for module, names in modules.items():
  63. mod = module.replace('.', '__')
  64. for name in names:
  65. test = 'test_' + mod + '__' + name
  66. if test not in ns:
  67. failed.append(module + '.' + name)
  68. assert not failed, "Missing classes: %s. Please add tests for these to sympy/core/tests/test_args.py." % ", ".join(failed)
  69. def _test_args(obj):
  70. all_basic = all(isinstance(arg, Basic) for arg in obj.args)
  71. # Ideally obj.func(*obj.args) would always recreate the object, but for
  72. # now, we only require it for objects with non-empty .args
  73. recreatable = not obj.args or obj.func(*obj.args) == obj
  74. return all_basic and recreatable
  75. def test_sympy__algebras__quaternion__Quaternion():
  76. from sympy.algebras.quaternion import Quaternion
  77. assert _test_args(Quaternion(x, 1, 2, 3))
  78. def test_sympy__assumptions__assume__AppliedPredicate():
  79. from sympy.assumptions.assume import AppliedPredicate, Predicate
  80. assert _test_args(AppliedPredicate(Predicate("test"), 2))
  81. assert _test_args(Q.is_true(True))
  82. @SKIP("abstract class")
  83. def test_sympy__assumptions__assume__Predicate():
  84. pass
  85. def test_predicates():
  86. predicates = [
  87. getattr(Q, attr)
  88. for attr in Q.__class__.__dict__
  89. if not attr.startswith('__')]
  90. for p in predicates:
  91. assert _test_args(p)
  92. def test_sympy__assumptions__assume__UndefinedPredicate():
  93. from sympy.assumptions.assume import Predicate
  94. assert _test_args(Predicate("test"))
  95. @SKIP('abstract class')
  96. def test_sympy__assumptions__relation__binrel__BinaryRelation():
  97. pass
  98. def test_sympy__assumptions__relation__binrel__AppliedBinaryRelation():
  99. assert _test_args(Q.eq(1, 2))
  100. def test_sympy__assumptions__wrapper__AssumptionsWrapper():
  101. from sympy.assumptions.wrapper import AssumptionsWrapper
  102. assert _test_args(AssumptionsWrapper(x, Q.positive(x)))
  103. @SKIP("abstract Class")
  104. def test_sympy__codegen__ast__CodegenAST():
  105. from sympy.codegen.ast import CodegenAST
  106. assert _test_args(CodegenAST())
  107. @SKIP("abstract Class")
  108. def test_sympy__codegen__ast__AssignmentBase():
  109. from sympy.codegen.ast import AssignmentBase
  110. assert _test_args(AssignmentBase(x, 1))
  111. @SKIP("abstract Class")
  112. def test_sympy__codegen__ast__AugmentedAssignment():
  113. from sympy.codegen.ast import AugmentedAssignment
  114. assert _test_args(AugmentedAssignment(x, 1))
  115. def test_sympy__codegen__ast__AddAugmentedAssignment():
  116. from sympy.codegen.ast import AddAugmentedAssignment
  117. assert _test_args(AddAugmentedAssignment(x, 1))
  118. def test_sympy__codegen__ast__SubAugmentedAssignment():
  119. from sympy.codegen.ast import SubAugmentedAssignment
  120. assert _test_args(SubAugmentedAssignment(x, 1))
  121. def test_sympy__codegen__ast__MulAugmentedAssignment():
  122. from sympy.codegen.ast import MulAugmentedAssignment
  123. assert _test_args(MulAugmentedAssignment(x, 1))
  124. def test_sympy__codegen__ast__DivAugmentedAssignment():
  125. from sympy.codegen.ast import DivAugmentedAssignment
  126. assert _test_args(DivAugmentedAssignment(x, 1))
  127. def test_sympy__codegen__ast__ModAugmentedAssignment():
  128. from sympy.codegen.ast import ModAugmentedAssignment
  129. assert _test_args(ModAugmentedAssignment(x, 1))
  130. def test_sympy__codegen__ast__CodeBlock():
  131. from sympy.codegen.ast import CodeBlock, Assignment
  132. assert _test_args(CodeBlock(Assignment(x, 1), Assignment(y, 2)))
  133. def test_sympy__codegen__ast__For():
  134. from sympy.codegen.ast import For, CodeBlock, AddAugmentedAssignment
  135. from sympy.sets import Range
  136. assert _test_args(For(x, Range(10), CodeBlock(AddAugmentedAssignment(y, 1))))
  137. def test_sympy__codegen__ast__Token():
  138. from sympy.codegen.ast import Token
  139. assert _test_args(Token())
  140. def test_sympy__codegen__ast__ContinueToken():
  141. from sympy.codegen.ast import ContinueToken
  142. assert _test_args(ContinueToken())
  143. def test_sympy__codegen__ast__BreakToken():
  144. from sympy.codegen.ast import BreakToken
  145. assert _test_args(BreakToken())
  146. def test_sympy__codegen__ast__NoneToken():
  147. from sympy.codegen.ast import NoneToken
  148. assert _test_args(NoneToken())
  149. def test_sympy__codegen__ast__String():
  150. from sympy.codegen.ast import String
  151. assert _test_args(String('foobar'))
  152. def test_sympy__codegen__ast__QuotedString():
  153. from sympy.codegen.ast import QuotedString
  154. assert _test_args(QuotedString('foobar'))
  155. def test_sympy__codegen__ast__Comment():
  156. from sympy.codegen.ast import Comment
  157. assert _test_args(Comment('this is a comment'))
  158. def test_sympy__codegen__ast__Node():
  159. from sympy.codegen.ast import Node
  160. assert _test_args(Node())
  161. assert _test_args(Node(attrs={1, 2, 3}))
  162. def test_sympy__codegen__ast__Type():
  163. from sympy.codegen.ast import Type
  164. assert _test_args(Type('float128'))
  165. def test_sympy__codegen__ast__IntBaseType():
  166. from sympy.codegen.ast import IntBaseType
  167. assert _test_args(IntBaseType('bigint'))
  168. def test_sympy__codegen__ast___SizedIntType():
  169. from sympy.codegen.ast import _SizedIntType
  170. assert _test_args(_SizedIntType('int128', 128))
  171. def test_sympy__codegen__ast__SignedIntType():
  172. from sympy.codegen.ast import SignedIntType
  173. assert _test_args(SignedIntType('int128_with_sign', 128))
  174. def test_sympy__codegen__ast__UnsignedIntType():
  175. from sympy.codegen.ast import UnsignedIntType
  176. assert _test_args(UnsignedIntType('unt128', 128))
  177. def test_sympy__codegen__ast__FloatBaseType():
  178. from sympy.codegen.ast import FloatBaseType
  179. assert _test_args(FloatBaseType('positive_real'))
  180. def test_sympy__codegen__ast__FloatType():
  181. from sympy.codegen.ast import FloatType
  182. assert _test_args(FloatType('float242', 242, nmant=142, nexp=99))
  183. def test_sympy__codegen__ast__ComplexBaseType():
  184. from sympy.codegen.ast import ComplexBaseType
  185. assert _test_args(ComplexBaseType('positive_cmplx'))
  186. def test_sympy__codegen__ast__ComplexType():
  187. from sympy.codegen.ast import ComplexType
  188. assert _test_args(ComplexType('complex42', 42, nmant=15, nexp=5))
  189. def test_sympy__codegen__ast__Attribute():
  190. from sympy.codegen.ast import Attribute
  191. assert _test_args(Attribute('noexcept'))
  192. def test_sympy__codegen__ast__Variable():
  193. from sympy.codegen.ast import Variable, Type, value_const
  194. assert _test_args(Variable(x))
  195. assert _test_args(Variable(y, Type('float32'), {value_const}))
  196. assert _test_args(Variable(z, type=Type('float64')))
  197. def test_sympy__codegen__ast__Pointer():
  198. from sympy.codegen.ast import Pointer, Type, pointer_const
  199. assert _test_args(Pointer(x))
  200. assert _test_args(Pointer(y, type=Type('float32')))
  201. assert _test_args(Pointer(z, Type('float64'), {pointer_const}))
  202. def test_sympy__codegen__ast__Declaration():
  203. from sympy.codegen.ast import Declaration, Variable, Type
  204. vx = Variable(x, type=Type('float'))
  205. assert _test_args(Declaration(vx))
  206. def test_sympy__codegen__ast__While():
  207. from sympy.codegen.ast import While, AddAugmentedAssignment
  208. assert _test_args(While(abs(x) < 1, [AddAugmentedAssignment(x, -1)]))
  209. def test_sympy__codegen__ast__Scope():
  210. from sympy.codegen.ast import Scope, AddAugmentedAssignment
  211. assert _test_args(Scope([AddAugmentedAssignment(x, -1)]))
  212. def test_sympy__codegen__ast__Stream():
  213. from sympy.codegen.ast import Stream
  214. assert _test_args(Stream('stdin'))
  215. def test_sympy__codegen__ast__Print():
  216. from sympy.codegen.ast import Print
  217. assert _test_args(Print([x, y]))
  218. assert _test_args(Print([x, y], "%d %d"))
  219. def test_sympy__codegen__ast__FunctionPrototype():
  220. from sympy.codegen.ast import FunctionPrototype, real, Declaration, Variable
  221. inp_x = Declaration(Variable(x, type=real))
  222. assert _test_args(FunctionPrototype(real, 'pwer', [inp_x]))
  223. def test_sympy__codegen__ast__FunctionDefinition():
  224. from sympy.codegen.ast import FunctionDefinition, real, Declaration, Variable, Assignment
  225. inp_x = Declaration(Variable(x, type=real))
  226. assert _test_args(FunctionDefinition(real, 'pwer', [inp_x], [Assignment(x, x**2)]))
  227. def test_sympy__codegen__ast__Return():
  228. from sympy.codegen.ast import Return
  229. assert _test_args(Return(x))
  230. def test_sympy__codegen__ast__FunctionCall():
  231. from sympy.codegen.ast import FunctionCall
  232. assert _test_args(FunctionCall('pwer', [x]))
  233. def test_sympy__codegen__ast__Element():
  234. from sympy.codegen.ast import Element
  235. assert _test_args(Element('x', range(3)))
  236. def test_sympy__codegen__cnodes__CommaOperator():
  237. from sympy.codegen.cnodes import CommaOperator
  238. assert _test_args(CommaOperator(1, 2))
  239. def test_sympy__codegen__cnodes__goto():
  240. from sympy.codegen.cnodes import goto
  241. assert _test_args(goto('early_exit'))
  242. def test_sympy__codegen__cnodes__Label():
  243. from sympy.codegen.cnodes import Label
  244. assert _test_args(Label('early_exit'))
  245. def test_sympy__codegen__cnodes__PreDecrement():
  246. from sympy.codegen.cnodes import PreDecrement
  247. assert _test_args(PreDecrement(x))
  248. def test_sympy__codegen__cnodes__PostDecrement():
  249. from sympy.codegen.cnodes import PostDecrement
  250. assert _test_args(PostDecrement(x))
  251. def test_sympy__codegen__cnodes__PreIncrement():
  252. from sympy.codegen.cnodes import PreIncrement
  253. assert _test_args(PreIncrement(x))
  254. def test_sympy__codegen__cnodes__PostIncrement():
  255. from sympy.codegen.cnodes import PostIncrement
  256. assert _test_args(PostIncrement(x))
  257. def test_sympy__codegen__cnodes__struct():
  258. from sympy.codegen.ast import real, Variable
  259. from sympy.codegen.cnodes import struct
  260. assert _test_args(struct(declarations=[
  261. Variable(x, type=real),
  262. Variable(y, type=real)
  263. ]))
  264. def test_sympy__codegen__cnodes__union():
  265. from sympy.codegen.ast import float32, int32, Variable
  266. from sympy.codegen.cnodes import union
  267. assert _test_args(union(declarations=[
  268. Variable(x, type=float32),
  269. Variable(y, type=int32)
  270. ]))
  271. def test_sympy__codegen__cxxnodes__using():
  272. from sympy.codegen.cxxnodes import using
  273. assert _test_args(using('std::vector'))
  274. assert _test_args(using('std::vector', 'vec'))
  275. def test_sympy__codegen__fnodes__Program():
  276. from sympy.codegen.fnodes import Program
  277. assert _test_args(Program('foobar', []))
  278. def test_sympy__codegen__fnodes__Module():
  279. from sympy.codegen.fnodes import Module
  280. assert _test_args(Module('foobar', [], []))
  281. def test_sympy__codegen__fnodes__Subroutine():
  282. from sympy.codegen.fnodes import Subroutine
  283. x = symbols('x', real=True)
  284. assert _test_args(Subroutine('foo', [x], []))
  285. def test_sympy__codegen__fnodes__GoTo():
  286. from sympy.codegen.fnodes import GoTo
  287. assert _test_args(GoTo([10]))
  288. assert _test_args(GoTo([10, 20], x > 1))
  289. def test_sympy__codegen__fnodes__FortranReturn():
  290. from sympy.codegen.fnodes import FortranReturn
  291. assert _test_args(FortranReturn(10))
  292. def test_sympy__codegen__fnodes__Extent():
  293. from sympy.codegen.fnodes import Extent
  294. assert _test_args(Extent())
  295. assert _test_args(Extent(None))
  296. assert _test_args(Extent(':'))
  297. assert _test_args(Extent(-3, 4))
  298. assert _test_args(Extent(x, y))
  299. def test_sympy__codegen__fnodes__use_rename():
  300. from sympy.codegen.fnodes import use_rename
  301. assert _test_args(use_rename('loc', 'glob'))
  302. def test_sympy__codegen__fnodes__use():
  303. from sympy.codegen.fnodes import use
  304. assert _test_args(use('modfoo', only='bar'))
  305. def test_sympy__codegen__fnodes__SubroutineCall():
  306. from sympy.codegen.fnodes import SubroutineCall
  307. assert _test_args(SubroutineCall('foo', ['bar', 'baz']))
  308. def test_sympy__codegen__fnodes__Do():
  309. from sympy.codegen.fnodes import Do
  310. assert _test_args(Do([], 'i', 1, 42))
  311. def test_sympy__codegen__fnodes__ImpliedDoLoop():
  312. from sympy.codegen.fnodes import ImpliedDoLoop
  313. assert _test_args(ImpliedDoLoop('i', 'i', 1, 42))
  314. def test_sympy__codegen__fnodes__ArrayConstructor():
  315. from sympy.codegen.fnodes import ArrayConstructor
  316. assert _test_args(ArrayConstructor([1, 2, 3]))
  317. from sympy.codegen.fnodes import ImpliedDoLoop
  318. idl = ImpliedDoLoop('i', 'i', 1, 42)
  319. assert _test_args(ArrayConstructor([1, idl, 3]))
  320. def test_sympy__codegen__fnodes__sum_():
  321. from sympy.codegen.fnodes import sum_
  322. assert _test_args(sum_('arr'))
  323. def test_sympy__codegen__fnodes__product_():
  324. from sympy.codegen.fnodes import product_
  325. assert _test_args(product_('arr'))
  326. def test_sympy__codegen__numpy_nodes__logaddexp():
  327. from sympy.codegen.numpy_nodes import logaddexp
  328. assert _test_args(logaddexp(x, y))
  329. def test_sympy__codegen__numpy_nodes__logaddexp2():
  330. from sympy.codegen.numpy_nodes import logaddexp2
  331. assert _test_args(logaddexp2(x, y))
  332. def test_sympy__codegen__pynodes__List():
  333. from sympy.codegen.pynodes import List
  334. assert _test_args(List(1, 2, 3))
  335. def test_sympy__codegen__pynodes__NumExprEvaluate():
  336. from sympy.codegen.pynodes import NumExprEvaluate
  337. assert _test_args(NumExprEvaluate(x))
  338. def test_sympy__codegen__scipy_nodes__cosm1():
  339. from sympy.codegen.scipy_nodes import cosm1
  340. assert _test_args(cosm1(x))
  341. def test_sympy__codegen__scipy_nodes__powm1():
  342. from sympy.codegen.scipy_nodes import powm1
  343. assert _test_args(powm1(x, y))
  344. def test_sympy__codegen__abstract_nodes__List():
  345. from sympy.codegen.abstract_nodes import List
  346. assert _test_args(List(1, 2, 3))
  347. def test_sympy__combinatorics__graycode__GrayCode():
  348. from sympy.combinatorics.graycode import GrayCode
  349. # an integer is given and returned from GrayCode as the arg
  350. assert _test_args(GrayCode(3, start='100'))
  351. assert _test_args(GrayCode(3, rank=1))
  352. def test_sympy__combinatorics__permutations__Permutation():
  353. from sympy.combinatorics.permutations import Permutation
  354. assert _test_args(Permutation([0, 1, 2, 3]))
  355. def test_sympy__combinatorics__permutations__AppliedPermutation():
  356. from sympy.combinatorics.permutations import Permutation
  357. from sympy.combinatorics.permutations import AppliedPermutation
  358. p = Permutation([0, 1, 2, 3])
  359. assert _test_args(AppliedPermutation(p, x))
  360. def test_sympy__combinatorics__perm_groups__PermutationGroup():
  361. from sympy.combinatorics.permutations import Permutation
  362. from sympy.combinatorics.perm_groups import PermutationGroup
  363. assert _test_args(PermutationGroup([Permutation([0, 1])]))
  364. def test_sympy__combinatorics__polyhedron__Polyhedron():
  365. from sympy.combinatorics.permutations import Permutation
  366. from sympy.combinatorics.polyhedron import Polyhedron
  367. from sympy.abc import w, x, y, z
  368. pgroup = [Permutation([[0, 1, 2], [3]]),
  369. Permutation([[0, 1, 3], [2]]),
  370. Permutation([[0, 2, 3], [1]]),
  371. Permutation([[1, 2, 3], [0]]),
  372. Permutation([[0, 1], [2, 3]]),
  373. Permutation([[0, 2], [1, 3]]),
  374. Permutation([[0, 3], [1, 2]]),
  375. Permutation([[0, 1, 2, 3]])]
  376. corners = [w, x, y, z]
  377. faces = [(w, x, y), (w, y, z), (w, z, x), (x, y, z)]
  378. assert _test_args(Polyhedron(corners, faces, pgroup))
  379. def test_sympy__combinatorics__prufer__Prufer():
  380. from sympy.combinatorics.prufer import Prufer
  381. assert _test_args(Prufer([[0, 1], [0, 2], [0, 3]], 4))
  382. def test_sympy__combinatorics__partitions__Partition():
  383. from sympy.combinatorics.partitions import Partition
  384. assert _test_args(Partition([1]))
  385. def test_sympy__combinatorics__partitions__IntegerPartition():
  386. from sympy.combinatorics.partitions import IntegerPartition
  387. assert _test_args(IntegerPartition([1]))
  388. def test_sympy__concrete__products__Product():
  389. from sympy.concrete.products import Product
  390. assert _test_args(Product(x, (x, 0, 10)))
  391. assert _test_args(Product(x, (x, 0, y), (y, 0, 10)))
  392. @SKIP("abstract Class")
  393. def test_sympy__concrete__expr_with_limits__ExprWithLimits():
  394. from sympy.concrete.expr_with_limits import ExprWithLimits
  395. assert _test_args(ExprWithLimits(x, (x, 0, 10)))
  396. assert _test_args(ExprWithLimits(x*y, (x, 0, 10.),(y,1.,3)))
  397. @SKIP("abstract Class")
  398. def test_sympy__concrete__expr_with_limits__AddWithLimits():
  399. from sympy.concrete.expr_with_limits import AddWithLimits
  400. assert _test_args(AddWithLimits(x, (x, 0, 10)))
  401. assert _test_args(AddWithLimits(x*y, (x, 0, 10),(y,1,3)))
  402. @SKIP("abstract Class")
  403. def test_sympy__concrete__expr_with_intlimits__ExprWithIntLimits():
  404. from sympy.concrete.expr_with_intlimits import ExprWithIntLimits
  405. assert _test_args(ExprWithIntLimits(x, (x, 0, 10)))
  406. assert _test_args(ExprWithIntLimits(x*y, (x, 0, 10),(y,1,3)))
  407. def test_sympy__concrete__summations__Sum():
  408. from sympy.concrete.summations import Sum
  409. assert _test_args(Sum(x, (x, 0, 10)))
  410. assert _test_args(Sum(x, (x, 0, y), (y, 0, 10)))
  411. def test_sympy__core__add__Add():
  412. from sympy.core.add import Add
  413. assert _test_args(Add(x, y, z, 2))
  414. def test_sympy__core__basic__Atom():
  415. from sympy.core.basic import Atom
  416. assert _test_args(Atom())
  417. def test_sympy__core__basic__Basic():
  418. from sympy.core.basic import Basic
  419. assert _test_args(Basic())
  420. def test_sympy__core__containers__Dict():
  421. from sympy.core.containers import Dict
  422. assert _test_args(Dict({x: y, y: z}))
  423. def test_sympy__core__containers__Tuple():
  424. from sympy.core.containers import Tuple
  425. assert _test_args(Tuple(x, y, z, 2))
  426. def test_sympy__core__expr__AtomicExpr():
  427. from sympy.core.expr import AtomicExpr
  428. assert _test_args(AtomicExpr())
  429. def test_sympy__core__expr__Expr():
  430. from sympy.core.expr import Expr
  431. assert _test_args(Expr())
  432. def test_sympy__core__expr__UnevaluatedExpr():
  433. from sympy.core.expr import UnevaluatedExpr
  434. from sympy.abc import x
  435. assert _test_args(UnevaluatedExpr(x))
  436. def test_sympy__core__function__Application():
  437. from sympy.core.function import Application
  438. assert _test_args(Application(1, 2, 3))
  439. def test_sympy__core__function__AppliedUndef():
  440. from sympy.core.function import AppliedUndef
  441. assert _test_args(AppliedUndef(1, 2, 3))
  442. def test_sympy__core__function__Derivative():
  443. from sympy.core.function import Derivative
  444. assert _test_args(Derivative(2, x, y, 3))
  445. @SKIP("abstract class")
  446. def test_sympy__core__function__Function():
  447. pass
  448. def test_sympy__core__function__Lambda():
  449. assert _test_args(Lambda((x, y), x + y + z))
  450. def test_sympy__core__function__Subs():
  451. from sympy.core.function import Subs
  452. assert _test_args(Subs(x + y, x, 2))
  453. def test_sympy__core__function__WildFunction():
  454. from sympy.core.function import WildFunction
  455. assert _test_args(WildFunction('f'))
  456. def test_sympy__core__mod__Mod():
  457. from sympy.core.mod import Mod
  458. assert _test_args(Mod(x, 2))
  459. def test_sympy__core__mul__Mul():
  460. from sympy.core.mul import Mul
  461. assert _test_args(Mul(2, x, y, z))
  462. def test_sympy__core__numbers__Catalan():
  463. from sympy.core.numbers import Catalan
  464. assert _test_args(Catalan())
  465. def test_sympy__core__numbers__ComplexInfinity():
  466. from sympy.core.numbers import ComplexInfinity
  467. assert _test_args(ComplexInfinity())
  468. def test_sympy__core__numbers__EulerGamma():
  469. from sympy.core.numbers import EulerGamma
  470. assert _test_args(EulerGamma())
  471. def test_sympy__core__numbers__Exp1():
  472. from sympy.core.numbers import Exp1
  473. assert _test_args(Exp1())
  474. def test_sympy__core__numbers__Float():
  475. from sympy.core.numbers import Float
  476. assert _test_args(Float(1.23))
  477. def test_sympy__core__numbers__GoldenRatio():
  478. from sympy.core.numbers import GoldenRatio
  479. assert _test_args(GoldenRatio())
  480. def test_sympy__core__numbers__TribonacciConstant():
  481. from sympy.core.numbers import TribonacciConstant
  482. assert _test_args(TribonacciConstant())
  483. def test_sympy__core__numbers__Half():
  484. from sympy.core.numbers import Half
  485. assert _test_args(Half())
  486. def test_sympy__core__numbers__ImaginaryUnit():
  487. from sympy.core.numbers import ImaginaryUnit
  488. assert _test_args(ImaginaryUnit())
  489. def test_sympy__core__numbers__Infinity():
  490. from sympy.core.numbers import Infinity
  491. assert _test_args(Infinity())
  492. def test_sympy__core__numbers__Integer():
  493. from sympy.core.numbers import Integer
  494. assert _test_args(Integer(7))
  495. @SKIP("abstract class")
  496. def test_sympy__core__numbers__IntegerConstant():
  497. pass
  498. def test_sympy__core__numbers__NaN():
  499. from sympy.core.numbers import NaN
  500. assert _test_args(NaN())
  501. def test_sympy__core__numbers__NegativeInfinity():
  502. from sympy.core.numbers import NegativeInfinity
  503. assert _test_args(NegativeInfinity())
  504. def test_sympy__core__numbers__NegativeOne():
  505. from sympy.core.numbers import NegativeOne
  506. assert _test_args(NegativeOne())
  507. def test_sympy__core__numbers__Number():
  508. from sympy.core.numbers import Number
  509. assert _test_args(Number(1, 7))
  510. def test_sympy__core__numbers__NumberSymbol():
  511. from sympy.core.numbers import NumberSymbol
  512. assert _test_args(NumberSymbol())
  513. def test_sympy__core__numbers__One():
  514. from sympy.core.numbers import One
  515. assert _test_args(One())
  516. def test_sympy__core__numbers__Pi():
  517. from sympy.core.numbers import Pi
  518. assert _test_args(Pi())
  519. def test_sympy__core__numbers__Rational():
  520. from sympy.core.numbers import Rational
  521. assert _test_args(Rational(1, 7))
  522. @SKIP("abstract class")
  523. def test_sympy__core__numbers__RationalConstant():
  524. pass
  525. def test_sympy__core__numbers__Zero():
  526. from sympy.core.numbers import Zero
  527. assert _test_args(Zero())
  528. @SKIP("abstract class")
  529. def test_sympy__core__operations__AssocOp():
  530. pass
  531. @SKIP("abstract class")
  532. def test_sympy__core__operations__LatticeOp():
  533. pass
  534. def test_sympy__core__power__Pow():
  535. from sympy.core.power import Pow
  536. assert _test_args(Pow(x, 2))
  537. def test_sympy__core__relational__Equality():
  538. from sympy.core.relational import Equality
  539. assert _test_args(Equality(x, 2))
  540. def test_sympy__core__relational__GreaterThan():
  541. from sympy.core.relational import GreaterThan
  542. assert _test_args(GreaterThan(x, 2))
  543. def test_sympy__core__relational__LessThan():
  544. from sympy.core.relational import LessThan
  545. assert _test_args(LessThan(x, 2))
  546. @SKIP("abstract class")
  547. def test_sympy__core__relational__Relational():
  548. pass
  549. def test_sympy__core__relational__StrictGreaterThan():
  550. from sympy.core.relational import StrictGreaterThan
  551. assert _test_args(StrictGreaterThan(x, 2))
  552. def test_sympy__core__relational__StrictLessThan():
  553. from sympy.core.relational import StrictLessThan
  554. assert _test_args(StrictLessThan(x, 2))
  555. def test_sympy__core__relational__Unequality():
  556. from sympy.core.relational import Unequality
  557. assert _test_args(Unequality(x, 2))
  558. def test_sympy__sandbox__indexed_integrals__IndexedIntegral():
  559. from sympy.tensor import IndexedBase, Idx
  560. from sympy.sandbox.indexed_integrals import IndexedIntegral
  561. A = IndexedBase('A')
  562. i, j = symbols('i j', integer=True)
  563. a1, a2 = symbols('a1:3', cls=Idx)
  564. assert _test_args(IndexedIntegral(A[a1], A[a2]))
  565. assert _test_args(IndexedIntegral(A[i], A[j]))
  566. def test_sympy__calculus__accumulationbounds__AccumulationBounds():
  567. from sympy.calculus.accumulationbounds import AccumulationBounds
  568. assert _test_args(AccumulationBounds(0, 1))
  569. def test_sympy__sets__ordinals__OmegaPower():
  570. from sympy.sets.ordinals import OmegaPower
  571. assert _test_args(OmegaPower(1, 1))
  572. def test_sympy__sets__ordinals__Ordinal():
  573. from sympy.sets.ordinals import Ordinal, OmegaPower
  574. assert _test_args(Ordinal(OmegaPower(2, 1)))
  575. def test_sympy__sets__ordinals__OrdinalOmega():
  576. from sympy.sets.ordinals import OrdinalOmega
  577. assert _test_args(OrdinalOmega())
  578. def test_sympy__sets__ordinals__OrdinalZero():
  579. from sympy.sets.ordinals import OrdinalZero
  580. assert _test_args(OrdinalZero())
  581. def test_sympy__sets__powerset__PowerSet():
  582. from sympy.sets.powerset import PowerSet
  583. from sympy.core.singleton import S
  584. assert _test_args(PowerSet(S.EmptySet))
  585. def test_sympy__sets__sets__EmptySet():
  586. from sympy.sets.sets import EmptySet
  587. assert _test_args(EmptySet())
  588. def test_sympy__sets__sets__UniversalSet():
  589. from sympy.sets.sets import UniversalSet
  590. assert _test_args(UniversalSet())
  591. def test_sympy__sets__sets__FiniteSet():
  592. from sympy.sets.sets import FiniteSet
  593. assert _test_args(FiniteSet(x, y, z))
  594. def test_sympy__sets__sets__Interval():
  595. from sympy.sets.sets import Interval
  596. assert _test_args(Interval(0, 1))
  597. def test_sympy__sets__sets__ProductSet():
  598. from sympy.sets.sets import ProductSet, Interval
  599. assert _test_args(ProductSet(Interval(0, 1), Interval(0, 1)))
  600. @SKIP("does it make sense to test this?")
  601. def test_sympy__sets__sets__Set():
  602. from sympy.sets.sets import Set
  603. assert _test_args(Set())
  604. def test_sympy__sets__sets__Intersection():
  605. from sympy.sets.sets import Intersection, Interval
  606. from sympy.core.symbol import Symbol
  607. x = Symbol('x')
  608. y = Symbol('y')
  609. S = Intersection(Interval(0, x), Interval(y, 1))
  610. assert isinstance(S, Intersection)
  611. assert _test_args(S)
  612. def test_sympy__sets__sets__Union():
  613. from sympy.sets.sets import Union, Interval
  614. assert _test_args(Union(Interval(0, 1), Interval(2, 3)))
  615. def test_sympy__sets__sets__Complement():
  616. from sympy.sets.sets import Complement, Interval
  617. assert _test_args(Complement(Interval(0, 2), Interval(0, 1)))
  618. def test_sympy__sets__sets__SymmetricDifference():
  619. from sympy.sets.sets import FiniteSet, SymmetricDifference
  620. assert _test_args(SymmetricDifference(FiniteSet(1, 2, 3), \
  621. FiniteSet(2, 3, 4)))
  622. def test_sympy__sets__sets__DisjointUnion():
  623. from sympy.sets.sets import FiniteSet, DisjointUnion
  624. assert _test_args(DisjointUnion(FiniteSet(1, 2, 3), \
  625. FiniteSet(2, 3, 4)))
  626. def test_sympy__physics__quantum__trace__Tr():
  627. from sympy.physics.quantum.trace import Tr
  628. a, b = symbols('a b', commutative=False)
  629. assert _test_args(Tr(a + b))
  630. def test_sympy__sets__setexpr__SetExpr():
  631. from sympy.sets.setexpr import SetExpr
  632. from sympy.sets.sets import Interval
  633. assert _test_args(SetExpr(Interval(0, 1)))
  634. def test_sympy__sets__fancysets__Rationals():
  635. from sympy.sets.fancysets import Rationals
  636. assert _test_args(Rationals())
  637. def test_sympy__sets__fancysets__Naturals():
  638. from sympy.sets.fancysets import Naturals
  639. assert _test_args(Naturals())
  640. def test_sympy__sets__fancysets__Naturals0():
  641. from sympy.sets.fancysets import Naturals0
  642. assert _test_args(Naturals0())
  643. def test_sympy__sets__fancysets__Integers():
  644. from sympy.sets.fancysets import Integers
  645. assert _test_args(Integers())
  646. def test_sympy__sets__fancysets__Reals():
  647. from sympy.sets.fancysets import Reals
  648. assert _test_args(Reals())
  649. def test_sympy__sets__fancysets__Complexes():
  650. from sympy.sets.fancysets import Complexes
  651. assert _test_args(Complexes())
  652. def test_sympy__sets__fancysets__ComplexRegion():
  653. from sympy.sets.fancysets import ComplexRegion
  654. from sympy.core.singleton import S
  655. from sympy.sets import Interval
  656. a = Interval(0, 1)
  657. b = Interval(2, 3)
  658. theta = Interval(0, 2*S.Pi)
  659. assert _test_args(ComplexRegion(a*b))
  660. assert _test_args(ComplexRegion(a*theta, polar=True))
  661. def test_sympy__sets__fancysets__CartesianComplexRegion():
  662. from sympy.sets.fancysets import CartesianComplexRegion
  663. from sympy.sets import Interval
  664. a = Interval(0, 1)
  665. b = Interval(2, 3)
  666. assert _test_args(CartesianComplexRegion(a*b))
  667. def test_sympy__sets__fancysets__PolarComplexRegion():
  668. from sympy.sets.fancysets import PolarComplexRegion
  669. from sympy.core.singleton import S
  670. from sympy.sets import Interval
  671. a = Interval(0, 1)
  672. theta = Interval(0, 2*S.Pi)
  673. assert _test_args(PolarComplexRegion(a*theta))
  674. def test_sympy__sets__fancysets__ImageSet():
  675. from sympy.sets.fancysets import ImageSet
  676. from sympy.core.singleton import S
  677. from sympy.core.symbol import Symbol
  678. x = Symbol('x')
  679. assert _test_args(ImageSet(Lambda(x, x**2), S.Naturals))
  680. def test_sympy__sets__fancysets__Range():
  681. from sympy.sets.fancysets import Range
  682. assert _test_args(Range(1, 5, 1))
  683. def test_sympy__sets__conditionset__ConditionSet():
  684. from sympy.sets.conditionset import ConditionSet
  685. from sympy.core.singleton import S
  686. from sympy.core.symbol import Symbol
  687. x = Symbol('x')
  688. assert _test_args(ConditionSet(x, Eq(x**2, 1), S.Reals))
  689. def test_sympy__sets__contains__Contains():
  690. from sympy.sets.fancysets import Range
  691. from sympy.sets.contains import Contains
  692. assert _test_args(Contains(x, Range(0, 10, 2)))
  693. # STATS
  694. from sympy.stats.crv_types import NormalDistribution
  695. nd = NormalDistribution(0, 1)
  696. from sympy.stats.frv_types import DieDistribution
  697. die = DieDistribution(6)
  698. def test_sympy__stats__crv__ContinuousDomain():
  699. from sympy.sets.sets import Interval
  700. from sympy.stats.crv import ContinuousDomain
  701. assert _test_args(ContinuousDomain({x}, Interval(-oo, oo)))
  702. def test_sympy__stats__crv__SingleContinuousDomain():
  703. from sympy.sets.sets import Interval
  704. from sympy.stats.crv import SingleContinuousDomain
  705. assert _test_args(SingleContinuousDomain(x, Interval(-oo, oo)))
  706. def test_sympy__stats__crv__ProductContinuousDomain():
  707. from sympy.sets.sets import Interval
  708. from sympy.stats.crv import SingleContinuousDomain, ProductContinuousDomain
  709. D = SingleContinuousDomain(x, Interval(-oo, oo))
  710. E = SingleContinuousDomain(y, Interval(0, oo))
  711. assert _test_args(ProductContinuousDomain(D, E))
  712. def test_sympy__stats__crv__ConditionalContinuousDomain():
  713. from sympy.sets.sets import Interval
  714. from sympy.stats.crv import (SingleContinuousDomain,
  715. ConditionalContinuousDomain)
  716. D = SingleContinuousDomain(x, Interval(-oo, oo))
  717. assert _test_args(ConditionalContinuousDomain(D, x > 0))
  718. def test_sympy__stats__crv__ContinuousPSpace():
  719. from sympy.sets.sets import Interval
  720. from sympy.stats.crv import ContinuousPSpace, SingleContinuousDomain
  721. D = SingleContinuousDomain(x, Interval(-oo, oo))
  722. assert _test_args(ContinuousPSpace(D, nd))
  723. def test_sympy__stats__crv__SingleContinuousPSpace():
  724. from sympy.stats.crv import SingleContinuousPSpace
  725. assert _test_args(SingleContinuousPSpace(x, nd))
  726. @SKIP("abstract class")
  727. def test_sympy__stats__rv__Distribution():
  728. pass
  729. @SKIP("abstract class")
  730. def test_sympy__stats__crv__SingleContinuousDistribution():
  731. pass
  732. def test_sympy__stats__drv__SingleDiscreteDomain():
  733. from sympy.stats.drv import SingleDiscreteDomain
  734. assert _test_args(SingleDiscreteDomain(x, S.Naturals))
  735. def test_sympy__stats__drv__ProductDiscreteDomain():
  736. from sympy.stats.drv import SingleDiscreteDomain, ProductDiscreteDomain
  737. X = SingleDiscreteDomain(x, S.Naturals)
  738. Y = SingleDiscreteDomain(y, S.Integers)
  739. assert _test_args(ProductDiscreteDomain(X, Y))
  740. def test_sympy__stats__drv__SingleDiscretePSpace():
  741. from sympy.stats.drv import SingleDiscretePSpace
  742. from sympy.stats.drv_types import PoissonDistribution
  743. assert _test_args(SingleDiscretePSpace(x, PoissonDistribution(1)))
  744. def test_sympy__stats__drv__DiscretePSpace():
  745. from sympy.stats.drv import DiscretePSpace, SingleDiscreteDomain
  746. density = Lambda(x, 2**(-x))
  747. domain = SingleDiscreteDomain(x, S.Naturals)
  748. assert _test_args(DiscretePSpace(domain, density))
  749. def test_sympy__stats__drv__ConditionalDiscreteDomain():
  750. from sympy.stats.drv import ConditionalDiscreteDomain, SingleDiscreteDomain
  751. X = SingleDiscreteDomain(x, S.Naturals0)
  752. assert _test_args(ConditionalDiscreteDomain(X, x > 2))
  753. def test_sympy__stats__joint_rv__JointPSpace():
  754. from sympy.stats.joint_rv import JointPSpace, JointDistribution
  755. assert _test_args(JointPSpace('X', JointDistribution(1)))
  756. def test_sympy__stats__joint_rv__JointRandomSymbol():
  757. from sympy.stats.joint_rv import JointRandomSymbol
  758. assert _test_args(JointRandomSymbol(x))
  759. def test_sympy__stats__joint_rv_types__JointDistributionHandmade():
  760. from sympy.tensor.indexed import Indexed
  761. from sympy.stats.joint_rv_types import JointDistributionHandmade
  762. x1, x2 = (Indexed('x', i) for i in (1, 2))
  763. assert _test_args(JointDistributionHandmade(x1 + x2, S.Reals**2))
  764. def test_sympy__stats__joint_rv__MarginalDistribution():
  765. from sympy.stats.rv import RandomSymbol
  766. from sympy.stats.joint_rv import MarginalDistribution
  767. r = RandomSymbol(S('r'))
  768. assert _test_args(MarginalDistribution(r, (r,)))
  769. def test_sympy__stats__compound_rv__CompoundDistribution():
  770. from sympy.stats.compound_rv import CompoundDistribution
  771. from sympy.stats.drv_types import PoissonDistribution, Poisson
  772. r = Poisson('r', 10)
  773. assert _test_args(CompoundDistribution(PoissonDistribution(r)))
  774. def test_sympy__stats__compound_rv__CompoundPSpace():
  775. from sympy.stats.compound_rv import CompoundPSpace, CompoundDistribution
  776. from sympy.stats.drv_types import PoissonDistribution, Poisson
  777. r = Poisson('r', 5)
  778. C = CompoundDistribution(PoissonDistribution(r))
  779. assert _test_args(CompoundPSpace('C', C))
  780. @SKIP("abstract class")
  781. def test_sympy__stats__drv__SingleDiscreteDistribution():
  782. pass
  783. @SKIP("abstract class")
  784. def test_sympy__stats__drv__DiscreteDistribution():
  785. pass
  786. @SKIP("abstract class")
  787. def test_sympy__stats__drv__DiscreteDomain():
  788. pass
  789. def test_sympy__stats__rv__RandomDomain():
  790. from sympy.stats.rv import RandomDomain
  791. from sympy.sets.sets import FiniteSet
  792. assert _test_args(RandomDomain(FiniteSet(x), FiniteSet(1, 2, 3)))
  793. def test_sympy__stats__rv__SingleDomain():
  794. from sympy.stats.rv import SingleDomain
  795. from sympy.sets.sets import FiniteSet
  796. assert _test_args(SingleDomain(x, FiniteSet(1, 2, 3)))
  797. def test_sympy__stats__rv__ConditionalDomain():
  798. from sympy.stats.rv import ConditionalDomain, RandomDomain
  799. from sympy.sets.sets import FiniteSet
  800. D = RandomDomain(FiniteSet(x), FiniteSet(1, 2))
  801. assert _test_args(ConditionalDomain(D, x > 1))
  802. def test_sympy__stats__rv__MatrixDomain():
  803. from sympy.stats.rv import MatrixDomain
  804. from sympy.matrices import MatrixSet
  805. from sympy.core.singleton import S
  806. assert _test_args(MatrixDomain(x, MatrixSet(2, 2, S.Reals)))
  807. def test_sympy__stats__rv__PSpace():
  808. from sympy.stats.rv import PSpace, RandomDomain
  809. from sympy.sets.sets import FiniteSet
  810. D = RandomDomain(FiniteSet(x), FiniteSet(1, 2, 3, 4, 5, 6))
  811. assert _test_args(PSpace(D, die))
  812. @SKIP("abstract Class")
  813. def test_sympy__stats__rv__SinglePSpace():
  814. pass
  815. def test_sympy__stats__rv__RandomSymbol():
  816. from sympy.stats.rv import RandomSymbol
  817. from sympy.stats.crv import SingleContinuousPSpace
  818. A = SingleContinuousPSpace(x, nd)
  819. assert _test_args(RandomSymbol(x, A))
  820. @SKIP("abstract Class")
  821. def test_sympy__stats__rv__ProductPSpace():
  822. pass
  823. def test_sympy__stats__rv__IndependentProductPSpace():
  824. from sympy.stats.rv import IndependentProductPSpace
  825. from sympy.stats.crv import SingleContinuousPSpace
  826. A = SingleContinuousPSpace(x, nd)
  827. B = SingleContinuousPSpace(y, nd)
  828. assert _test_args(IndependentProductPSpace(A, B))
  829. def test_sympy__stats__rv__ProductDomain():
  830. from sympy.sets.sets import Interval
  831. from sympy.stats.rv import ProductDomain, SingleDomain
  832. D = SingleDomain(x, Interval(-oo, oo))
  833. E = SingleDomain(y, Interval(0, oo))
  834. assert _test_args(ProductDomain(D, E))
  835. def test_sympy__stats__symbolic_probability__Probability():
  836. from sympy.stats.symbolic_probability import Probability
  837. from sympy.stats import Normal
  838. X = Normal('X', 0, 1)
  839. assert _test_args(Probability(X > 0))
  840. def test_sympy__stats__symbolic_probability__Expectation():
  841. from sympy.stats.symbolic_probability import Expectation
  842. from sympy.stats import Normal
  843. X = Normal('X', 0, 1)
  844. assert _test_args(Expectation(X > 0))
  845. def test_sympy__stats__symbolic_probability__Covariance():
  846. from sympy.stats.symbolic_probability import Covariance
  847. from sympy.stats import Normal
  848. X = Normal('X', 0, 1)
  849. Y = Normal('Y', 0, 3)
  850. assert _test_args(Covariance(X, Y))
  851. def test_sympy__stats__symbolic_probability__Variance():
  852. from sympy.stats.symbolic_probability import Variance
  853. from sympy.stats import Normal
  854. X = Normal('X', 0, 1)
  855. assert _test_args(Variance(X))
  856. def test_sympy__stats__symbolic_probability__Moment():
  857. from sympy.stats.symbolic_probability import Moment
  858. from sympy.stats import Normal
  859. X = Normal('X', 0, 1)
  860. assert _test_args(Moment(X, 3, 2, X > 3))
  861. def test_sympy__stats__symbolic_probability__CentralMoment():
  862. from sympy.stats.symbolic_probability import CentralMoment
  863. from sympy.stats import Normal
  864. X = Normal('X', 0, 1)
  865. assert _test_args(CentralMoment(X, 2, X > 1))
  866. def test_sympy__stats__frv_types__DiscreteUniformDistribution():
  867. from sympy.stats.frv_types import DiscreteUniformDistribution
  868. from sympy.core.containers import Tuple
  869. assert _test_args(DiscreteUniformDistribution(Tuple(*list(range(6)))))
  870. def test_sympy__stats__frv_types__DieDistribution():
  871. assert _test_args(die)
  872. def test_sympy__stats__frv_types__BernoulliDistribution():
  873. from sympy.stats.frv_types import BernoulliDistribution
  874. assert _test_args(BernoulliDistribution(S.Half, 0, 1))
  875. def test_sympy__stats__frv_types__BinomialDistribution():
  876. from sympy.stats.frv_types import BinomialDistribution
  877. assert _test_args(BinomialDistribution(5, S.Half, 1, 0))
  878. def test_sympy__stats__frv_types__BetaBinomialDistribution():
  879. from sympy.stats.frv_types import BetaBinomialDistribution
  880. assert _test_args(BetaBinomialDistribution(5, 1, 1))
  881. def test_sympy__stats__frv_types__HypergeometricDistribution():
  882. from sympy.stats.frv_types import HypergeometricDistribution
  883. assert _test_args(HypergeometricDistribution(10, 5, 3))
  884. def test_sympy__stats__frv_types__RademacherDistribution():
  885. from sympy.stats.frv_types import RademacherDistribution
  886. assert _test_args(RademacherDistribution())
  887. def test_sympy__stats__frv_types__IdealSolitonDistribution():
  888. from sympy.stats.frv_types import IdealSolitonDistribution
  889. assert _test_args(IdealSolitonDistribution(10))
  890. def test_sympy__stats__frv_types__RobustSolitonDistribution():
  891. from sympy.stats.frv_types import RobustSolitonDistribution
  892. assert _test_args(RobustSolitonDistribution(1000, 0.5, 0.1))
  893. def test_sympy__stats__frv__FiniteDomain():
  894. from sympy.stats.frv import FiniteDomain
  895. assert _test_args(FiniteDomain({(x, 1), (x, 2)})) # x can be 1 or 2
  896. def test_sympy__stats__frv__SingleFiniteDomain():
  897. from sympy.stats.frv import SingleFiniteDomain
  898. assert _test_args(SingleFiniteDomain(x, {1, 2})) # x can be 1 or 2
  899. def test_sympy__stats__frv__ProductFiniteDomain():
  900. from sympy.stats.frv import SingleFiniteDomain, ProductFiniteDomain
  901. xd = SingleFiniteDomain(x, {1, 2})
  902. yd = SingleFiniteDomain(y, {1, 2})
  903. assert _test_args(ProductFiniteDomain(xd, yd))
  904. def test_sympy__stats__frv__ConditionalFiniteDomain():
  905. from sympy.stats.frv import SingleFiniteDomain, ConditionalFiniteDomain
  906. xd = SingleFiniteDomain(x, {1, 2})
  907. assert _test_args(ConditionalFiniteDomain(xd, x > 1))
  908. def test_sympy__stats__frv__FinitePSpace():
  909. from sympy.stats.frv import FinitePSpace, SingleFiniteDomain
  910. xd = SingleFiniteDomain(x, {1, 2, 3, 4, 5, 6})
  911. assert _test_args(FinitePSpace(xd, {(x, 1): S.Half, (x, 2): S.Half}))
  912. xd = SingleFiniteDomain(x, {1, 2})
  913. assert _test_args(FinitePSpace(xd, {(x, 1): S.Half, (x, 2): S.Half}))
  914. def test_sympy__stats__frv__SingleFinitePSpace():
  915. from sympy.stats.frv import SingleFinitePSpace
  916. from sympy.core.symbol import Symbol
  917. assert _test_args(SingleFinitePSpace(Symbol('x'), die))
  918. def test_sympy__stats__frv__ProductFinitePSpace():
  919. from sympy.stats.frv import SingleFinitePSpace, ProductFinitePSpace
  920. from sympy.core.symbol import Symbol
  921. xp = SingleFinitePSpace(Symbol('x'), die)
  922. yp = SingleFinitePSpace(Symbol('y'), die)
  923. assert _test_args(ProductFinitePSpace(xp, yp))
  924. @SKIP("abstract class")
  925. def test_sympy__stats__frv__SingleFiniteDistribution():
  926. pass
  927. @SKIP("abstract class")
  928. def test_sympy__stats__crv__ContinuousDistribution():
  929. pass
  930. def test_sympy__stats__frv_types__FiniteDistributionHandmade():
  931. from sympy.stats.frv_types import FiniteDistributionHandmade
  932. from sympy.core.containers import Dict
  933. assert _test_args(FiniteDistributionHandmade(Dict({1: 1})))
  934. def test_sympy__stats__crv_types__ContinuousDistributionHandmade():
  935. from sympy.stats.crv_types import ContinuousDistributionHandmade
  936. from sympy.core.function import Lambda
  937. from sympy.sets.sets import Interval
  938. from sympy.abc import x
  939. assert _test_args(ContinuousDistributionHandmade(Lambda(x, 2*x),
  940. Interval(0, 1)))
  941. def test_sympy__stats__drv_types__DiscreteDistributionHandmade():
  942. from sympy.stats.drv_types import DiscreteDistributionHandmade
  943. from sympy.core.function import Lambda
  944. from sympy.sets.sets import FiniteSet
  945. from sympy.abc import x
  946. assert _test_args(DiscreteDistributionHandmade(Lambda(x, Rational(1, 10)),
  947. FiniteSet(*range(10))))
  948. def test_sympy__stats__rv__Density():
  949. from sympy.stats.rv import Density
  950. from sympy.stats.crv_types import Normal
  951. assert _test_args(Density(Normal('x', 0, 1)))
  952. def test_sympy__stats__crv_types__ArcsinDistribution():
  953. from sympy.stats.crv_types import ArcsinDistribution
  954. assert _test_args(ArcsinDistribution(0, 1))
  955. def test_sympy__stats__crv_types__BeniniDistribution():
  956. from sympy.stats.crv_types import BeniniDistribution
  957. assert _test_args(BeniniDistribution(1, 1, 1))
  958. def test_sympy__stats__crv_types__BetaDistribution():
  959. from sympy.stats.crv_types import BetaDistribution
  960. assert _test_args(BetaDistribution(1, 1))
  961. def test_sympy__stats__crv_types__BetaNoncentralDistribution():
  962. from sympy.stats.crv_types import BetaNoncentralDistribution
  963. assert _test_args(BetaNoncentralDistribution(1, 1, 1))
  964. def test_sympy__stats__crv_types__BetaPrimeDistribution():
  965. from sympy.stats.crv_types import BetaPrimeDistribution
  966. assert _test_args(BetaPrimeDistribution(1, 1))
  967. def test_sympy__stats__crv_types__BoundedParetoDistribution():
  968. from sympy.stats.crv_types import BoundedParetoDistribution
  969. assert _test_args(BoundedParetoDistribution(1, 1, 2))
  970. def test_sympy__stats__crv_types__CauchyDistribution():
  971. from sympy.stats.crv_types import CauchyDistribution
  972. assert _test_args(CauchyDistribution(0, 1))
  973. def test_sympy__stats__crv_types__ChiDistribution():
  974. from sympy.stats.crv_types import ChiDistribution
  975. assert _test_args(ChiDistribution(1))
  976. def test_sympy__stats__crv_types__ChiNoncentralDistribution():
  977. from sympy.stats.crv_types import ChiNoncentralDistribution
  978. assert _test_args(ChiNoncentralDistribution(1,1))
  979. def test_sympy__stats__crv_types__ChiSquaredDistribution():
  980. from sympy.stats.crv_types import ChiSquaredDistribution
  981. assert _test_args(ChiSquaredDistribution(1))
  982. def test_sympy__stats__crv_types__DagumDistribution():
  983. from sympy.stats.crv_types import DagumDistribution
  984. assert _test_args(DagumDistribution(1, 1, 1))
  985. def test_sympy__stats__crv_types__ExGaussianDistribution():
  986. from sympy.stats.crv_types import ExGaussianDistribution
  987. assert _test_args(ExGaussianDistribution(1, 1, 1))
  988. def test_sympy__stats__crv_types__ExponentialDistribution():
  989. from sympy.stats.crv_types import ExponentialDistribution
  990. assert _test_args(ExponentialDistribution(1))
  991. def test_sympy__stats__crv_types__ExponentialPowerDistribution():
  992. from sympy.stats.crv_types import ExponentialPowerDistribution
  993. assert _test_args(ExponentialPowerDistribution(0, 1, 1))
  994. def test_sympy__stats__crv_types__FDistributionDistribution():
  995. from sympy.stats.crv_types import FDistributionDistribution
  996. assert _test_args(FDistributionDistribution(1, 1))
  997. def test_sympy__stats__crv_types__FisherZDistribution():
  998. from sympy.stats.crv_types import FisherZDistribution
  999. assert _test_args(FisherZDistribution(1, 1))
  1000. def test_sympy__stats__crv_types__FrechetDistribution():
  1001. from sympy.stats.crv_types import FrechetDistribution
  1002. assert _test_args(FrechetDistribution(1, 1, 1))
  1003. def test_sympy__stats__crv_types__GammaInverseDistribution():
  1004. from sympy.stats.crv_types import GammaInverseDistribution
  1005. assert _test_args(GammaInverseDistribution(1, 1))
  1006. def test_sympy__stats__crv_types__GammaDistribution():
  1007. from sympy.stats.crv_types import GammaDistribution
  1008. assert _test_args(GammaDistribution(1, 1))
  1009. def test_sympy__stats__crv_types__GumbelDistribution():
  1010. from sympy.stats.crv_types import GumbelDistribution
  1011. assert _test_args(GumbelDistribution(1, 1, False))
  1012. def test_sympy__stats__crv_types__GompertzDistribution():
  1013. from sympy.stats.crv_types import GompertzDistribution
  1014. assert _test_args(GompertzDistribution(1, 1))
  1015. def test_sympy__stats__crv_types__KumaraswamyDistribution():
  1016. from sympy.stats.crv_types import KumaraswamyDistribution
  1017. assert _test_args(KumaraswamyDistribution(1, 1))
  1018. def test_sympy__stats__crv_types__LaplaceDistribution():
  1019. from sympy.stats.crv_types import LaplaceDistribution
  1020. assert _test_args(LaplaceDistribution(0, 1))
  1021. def test_sympy__stats__crv_types__LevyDistribution():
  1022. from sympy.stats.crv_types import LevyDistribution
  1023. assert _test_args(LevyDistribution(0, 1))
  1024. def test_sympy__stats__crv_types__LogCauchyDistribution():
  1025. from sympy.stats.crv_types import LogCauchyDistribution
  1026. assert _test_args(LogCauchyDistribution(0, 1))
  1027. def test_sympy__stats__crv_types__LogisticDistribution():
  1028. from sympy.stats.crv_types import LogisticDistribution
  1029. assert _test_args(LogisticDistribution(0, 1))
  1030. def test_sympy__stats__crv_types__LogLogisticDistribution():
  1031. from sympy.stats.crv_types import LogLogisticDistribution
  1032. assert _test_args(LogLogisticDistribution(1, 1))
  1033. def test_sympy__stats__crv_types__LogitNormalDistribution():
  1034. from sympy.stats.crv_types import LogitNormalDistribution
  1035. assert _test_args(LogitNormalDistribution(0, 1))
  1036. def test_sympy__stats__crv_types__LogNormalDistribution():
  1037. from sympy.stats.crv_types import LogNormalDistribution
  1038. assert _test_args(LogNormalDistribution(0, 1))
  1039. def test_sympy__stats__crv_types__LomaxDistribution():
  1040. from sympy.stats.crv_types import LomaxDistribution
  1041. assert _test_args(LomaxDistribution(1, 2))
  1042. def test_sympy__stats__crv_types__MaxwellDistribution():
  1043. from sympy.stats.crv_types import MaxwellDistribution
  1044. assert _test_args(MaxwellDistribution(1))
  1045. def test_sympy__stats__crv_types__MoyalDistribution():
  1046. from sympy.stats.crv_types import MoyalDistribution
  1047. assert _test_args(MoyalDistribution(1,2))
  1048. def test_sympy__stats__crv_types__NakagamiDistribution():
  1049. from sympy.stats.crv_types import NakagamiDistribution
  1050. assert _test_args(NakagamiDistribution(1, 1))
  1051. def test_sympy__stats__crv_types__NormalDistribution():
  1052. from sympy.stats.crv_types import NormalDistribution
  1053. assert _test_args(NormalDistribution(0, 1))
  1054. def test_sympy__stats__crv_types__GaussianInverseDistribution():
  1055. from sympy.stats.crv_types import GaussianInverseDistribution
  1056. assert _test_args(GaussianInverseDistribution(1, 1))
  1057. def test_sympy__stats__crv_types__ParetoDistribution():
  1058. from sympy.stats.crv_types import ParetoDistribution
  1059. assert _test_args(ParetoDistribution(1, 1))
  1060. def test_sympy__stats__crv_types__PowerFunctionDistribution():
  1061. from sympy.stats.crv_types import PowerFunctionDistribution
  1062. assert _test_args(PowerFunctionDistribution(2,0,1))
  1063. def test_sympy__stats__crv_types__QuadraticUDistribution():
  1064. from sympy.stats.crv_types import QuadraticUDistribution
  1065. assert _test_args(QuadraticUDistribution(1, 2))
  1066. def test_sympy__stats__crv_types__RaisedCosineDistribution():
  1067. from sympy.stats.crv_types import RaisedCosineDistribution
  1068. assert _test_args(RaisedCosineDistribution(1, 1))
  1069. def test_sympy__stats__crv_types__RayleighDistribution():
  1070. from sympy.stats.crv_types import RayleighDistribution
  1071. assert _test_args(RayleighDistribution(1))
  1072. def test_sympy__stats__crv_types__ReciprocalDistribution():
  1073. from sympy.stats.crv_types import ReciprocalDistribution
  1074. assert _test_args(ReciprocalDistribution(5, 30))
  1075. def test_sympy__stats__crv_types__ShiftedGompertzDistribution():
  1076. from sympy.stats.crv_types import ShiftedGompertzDistribution
  1077. assert _test_args(ShiftedGompertzDistribution(1, 1))
  1078. def test_sympy__stats__crv_types__StudentTDistribution():
  1079. from sympy.stats.crv_types import StudentTDistribution
  1080. assert _test_args(StudentTDistribution(1))
  1081. def test_sympy__stats__crv_types__TrapezoidalDistribution():
  1082. from sympy.stats.crv_types import TrapezoidalDistribution
  1083. assert _test_args(TrapezoidalDistribution(1, 2, 3, 4))
  1084. def test_sympy__stats__crv_types__TriangularDistribution():
  1085. from sympy.stats.crv_types import TriangularDistribution
  1086. assert _test_args(TriangularDistribution(-1, 0, 1))
  1087. def test_sympy__stats__crv_types__UniformDistribution():
  1088. from sympy.stats.crv_types import UniformDistribution
  1089. assert _test_args(UniformDistribution(0, 1))
  1090. def test_sympy__stats__crv_types__UniformSumDistribution():
  1091. from sympy.stats.crv_types import UniformSumDistribution
  1092. assert _test_args(UniformSumDistribution(1))
  1093. def test_sympy__stats__crv_types__VonMisesDistribution():
  1094. from sympy.stats.crv_types import VonMisesDistribution
  1095. assert _test_args(VonMisesDistribution(1, 1))
  1096. def test_sympy__stats__crv_types__WeibullDistribution():
  1097. from sympy.stats.crv_types import WeibullDistribution
  1098. assert _test_args(WeibullDistribution(1, 1))
  1099. def test_sympy__stats__crv_types__WignerSemicircleDistribution():
  1100. from sympy.stats.crv_types import WignerSemicircleDistribution
  1101. assert _test_args(WignerSemicircleDistribution(1))
  1102. def test_sympy__stats__drv_types__GeometricDistribution():
  1103. from sympy.stats.drv_types import GeometricDistribution
  1104. assert _test_args(GeometricDistribution(.5))
  1105. def test_sympy__stats__drv_types__HermiteDistribution():
  1106. from sympy.stats.drv_types import HermiteDistribution
  1107. assert _test_args(HermiteDistribution(1, 2))
  1108. def test_sympy__stats__drv_types__LogarithmicDistribution():
  1109. from sympy.stats.drv_types import LogarithmicDistribution
  1110. assert _test_args(LogarithmicDistribution(.5))
  1111. def test_sympy__stats__drv_types__NegativeBinomialDistribution():
  1112. from sympy.stats.drv_types import NegativeBinomialDistribution
  1113. assert _test_args(NegativeBinomialDistribution(.5, .5))
  1114. def test_sympy__stats__drv_types__FlorySchulzDistribution():
  1115. from sympy.stats.drv_types import FlorySchulzDistribution
  1116. assert _test_args(FlorySchulzDistribution(.5))
  1117. def test_sympy__stats__drv_types__PoissonDistribution():
  1118. from sympy.stats.drv_types import PoissonDistribution
  1119. assert _test_args(PoissonDistribution(1))
  1120. def test_sympy__stats__drv_types__SkellamDistribution():
  1121. from sympy.stats.drv_types import SkellamDistribution
  1122. assert _test_args(SkellamDistribution(1, 1))
  1123. def test_sympy__stats__drv_types__YuleSimonDistribution():
  1124. from sympy.stats.drv_types import YuleSimonDistribution
  1125. assert _test_args(YuleSimonDistribution(.5))
  1126. def test_sympy__stats__drv_types__ZetaDistribution():
  1127. from sympy.stats.drv_types import ZetaDistribution
  1128. assert _test_args(ZetaDistribution(1.5))
  1129. def test_sympy__stats__joint_rv__JointDistribution():
  1130. from sympy.stats.joint_rv import JointDistribution
  1131. assert _test_args(JointDistribution(1, 2, 3, 4))
  1132. def test_sympy__stats__joint_rv_types__MultivariateNormalDistribution():
  1133. from sympy.stats.joint_rv_types import MultivariateNormalDistribution
  1134. assert _test_args(
  1135. MultivariateNormalDistribution([0, 1], [[1, 0],[0, 1]]))
  1136. def test_sympy__stats__joint_rv_types__MultivariateLaplaceDistribution():
  1137. from sympy.stats.joint_rv_types import MultivariateLaplaceDistribution
  1138. assert _test_args(MultivariateLaplaceDistribution([0, 1], [[1, 0],[0, 1]]))
  1139. def test_sympy__stats__joint_rv_types__MultivariateTDistribution():
  1140. from sympy.stats.joint_rv_types import MultivariateTDistribution
  1141. assert _test_args(MultivariateTDistribution([0, 1], [[1, 0],[0, 1]], 1))
  1142. def test_sympy__stats__joint_rv_types__NormalGammaDistribution():
  1143. from sympy.stats.joint_rv_types import NormalGammaDistribution
  1144. assert _test_args(NormalGammaDistribution(1, 2, 3, 4))
  1145. def test_sympy__stats__joint_rv_types__GeneralizedMultivariateLogGammaDistribution():
  1146. from sympy.stats.joint_rv_types import GeneralizedMultivariateLogGammaDistribution
  1147. v, l, mu = (4, [1, 2, 3, 4], [1, 2, 3, 4])
  1148. assert _test_args(GeneralizedMultivariateLogGammaDistribution(S.Half, v, l, mu))
  1149. def test_sympy__stats__joint_rv_types__MultivariateBetaDistribution():
  1150. from sympy.stats.joint_rv_types import MultivariateBetaDistribution
  1151. assert _test_args(MultivariateBetaDistribution([1, 2, 3]))
  1152. def test_sympy__stats__joint_rv_types__MultivariateEwensDistribution():
  1153. from sympy.stats.joint_rv_types import MultivariateEwensDistribution
  1154. assert _test_args(MultivariateEwensDistribution(5, 1))
  1155. def test_sympy__stats__joint_rv_types__MultinomialDistribution():
  1156. from sympy.stats.joint_rv_types import MultinomialDistribution
  1157. assert _test_args(MultinomialDistribution(5, [0.5, 0.1, 0.3]))
  1158. def test_sympy__stats__joint_rv_types__NegativeMultinomialDistribution():
  1159. from sympy.stats.joint_rv_types import NegativeMultinomialDistribution
  1160. assert _test_args(NegativeMultinomialDistribution(5, [0.5, 0.1, 0.3]))
  1161. def test_sympy__stats__rv__RandomIndexedSymbol():
  1162. from sympy.stats.rv import RandomIndexedSymbol, pspace
  1163. from sympy.stats.stochastic_process_types import DiscreteMarkovChain
  1164. X = DiscreteMarkovChain("X")
  1165. assert _test_args(RandomIndexedSymbol(X[0].symbol, pspace(X[0])))
  1166. def test_sympy__stats__rv__RandomMatrixSymbol():
  1167. from sympy.stats.rv import RandomMatrixSymbol
  1168. from sympy.stats.random_matrix import RandomMatrixPSpace
  1169. pspace = RandomMatrixPSpace('P')
  1170. assert _test_args(RandomMatrixSymbol('M', 3, 3, pspace))
  1171. def test_sympy__stats__stochastic_process__StochasticPSpace():
  1172. from sympy.stats.stochastic_process import StochasticPSpace
  1173. from sympy.stats.stochastic_process_types import StochasticProcess
  1174. from sympy.stats.frv_types import BernoulliDistribution
  1175. assert _test_args(StochasticPSpace("Y", StochasticProcess("Y", [1, 2, 3]), BernoulliDistribution(S.Half, 1, 0)))
  1176. def test_sympy__stats__stochastic_process_types__StochasticProcess():
  1177. from sympy.stats.stochastic_process_types import StochasticProcess
  1178. assert _test_args(StochasticProcess("Y", [1, 2, 3]))
  1179. def test_sympy__stats__stochastic_process_types__MarkovProcess():
  1180. from sympy.stats.stochastic_process_types import MarkovProcess
  1181. assert _test_args(MarkovProcess("Y", [1, 2, 3]))
  1182. def test_sympy__stats__stochastic_process_types__DiscreteTimeStochasticProcess():
  1183. from sympy.stats.stochastic_process_types import DiscreteTimeStochasticProcess
  1184. assert _test_args(DiscreteTimeStochasticProcess("Y", [1, 2, 3]))
  1185. def test_sympy__stats__stochastic_process_types__ContinuousTimeStochasticProcess():
  1186. from sympy.stats.stochastic_process_types import ContinuousTimeStochasticProcess
  1187. assert _test_args(ContinuousTimeStochasticProcess("Y", [1, 2, 3]))
  1188. def test_sympy__stats__stochastic_process_types__TransitionMatrixOf():
  1189. from sympy.stats.stochastic_process_types import TransitionMatrixOf, DiscreteMarkovChain
  1190. from sympy.matrices.expressions.matexpr import MatrixSymbol
  1191. DMC = DiscreteMarkovChain("Y")
  1192. assert _test_args(TransitionMatrixOf(DMC, MatrixSymbol('T', 3, 3)))
  1193. def test_sympy__stats__stochastic_process_types__GeneratorMatrixOf():
  1194. from sympy.stats.stochastic_process_types import GeneratorMatrixOf, ContinuousMarkovChain
  1195. from sympy.matrices.expressions.matexpr import MatrixSymbol
  1196. DMC = ContinuousMarkovChain("Y")
  1197. assert _test_args(GeneratorMatrixOf(DMC, MatrixSymbol('T', 3, 3)))
  1198. def test_sympy__stats__stochastic_process_types__StochasticStateSpaceOf():
  1199. from sympy.stats.stochastic_process_types import StochasticStateSpaceOf, DiscreteMarkovChain
  1200. DMC = DiscreteMarkovChain("Y")
  1201. assert _test_args(StochasticStateSpaceOf(DMC, [0, 1, 2]))
  1202. def test_sympy__stats__stochastic_process_types__DiscreteMarkovChain():
  1203. from sympy.stats.stochastic_process_types import DiscreteMarkovChain
  1204. from sympy.matrices.expressions.matexpr import MatrixSymbol
  1205. assert _test_args(DiscreteMarkovChain("Y", [0, 1, 2], MatrixSymbol('T', 3, 3)))
  1206. def test_sympy__stats__stochastic_process_types__ContinuousMarkovChain():
  1207. from sympy.stats.stochastic_process_types import ContinuousMarkovChain
  1208. from sympy.matrices.expressions.matexpr import MatrixSymbol
  1209. assert _test_args(ContinuousMarkovChain("Y", [0, 1, 2], MatrixSymbol('T', 3, 3)))
  1210. def test_sympy__stats__stochastic_process_types__BernoulliProcess():
  1211. from sympy.stats.stochastic_process_types import BernoulliProcess
  1212. assert _test_args(BernoulliProcess("B", 0.5, 1, 0))
  1213. def test_sympy__stats__stochastic_process_types__CountingProcess():
  1214. from sympy.stats.stochastic_process_types import CountingProcess
  1215. assert _test_args(CountingProcess("C"))
  1216. def test_sympy__stats__stochastic_process_types__PoissonProcess():
  1217. from sympy.stats.stochastic_process_types import PoissonProcess
  1218. assert _test_args(PoissonProcess("X", 2))
  1219. def test_sympy__stats__stochastic_process_types__WienerProcess():
  1220. from sympy.stats.stochastic_process_types import WienerProcess
  1221. assert _test_args(WienerProcess("X"))
  1222. def test_sympy__stats__stochastic_process_types__GammaProcess():
  1223. from sympy.stats.stochastic_process_types import GammaProcess
  1224. assert _test_args(GammaProcess("X", 1, 2))
  1225. def test_sympy__stats__random_matrix__RandomMatrixPSpace():
  1226. from sympy.stats.random_matrix import RandomMatrixPSpace
  1227. from sympy.stats.random_matrix_models import RandomMatrixEnsembleModel
  1228. model = RandomMatrixEnsembleModel('R', 3)
  1229. assert _test_args(RandomMatrixPSpace('P', model=model))
  1230. def test_sympy__stats__random_matrix_models__RandomMatrixEnsembleModel():
  1231. from sympy.stats.random_matrix_models import RandomMatrixEnsembleModel
  1232. assert _test_args(RandomMatrixEnsembleModel('R', 3))
  1233. def test_sympy__stats__random_matrix_models__GaussianEnsembleModel():
  1234. from sympy.stats.random_matrix_models import GaussianEnsembleModel
  1235. assert _test_args(GaussianEnsembleModel('G', 3))
  1236. def test_sympy__stats__random_matrix_models__GaussianUnitaryEnsembleModel():
  1237. from sympy.stats.random_matrix_models import GaussianUnitaryEnsembleModel
  1238. assert _test_args(GaussianUnitaryEnsembleModel('U', 3))
  1239. def test_sympy__stats__random_matrix_models__GaussianOrthogonalEnsembleModel():
  1240. from sympy.stats.random_matrix_models import GaussianOrthogonalEnsembleModel
  1241. assert _test_args(GaussianOrthogonalEnsembleModel('U', 3))
  1242. def test_sympy__stats__random_matrix_models__GaussianSymplecticEnsembleModel():
  1243. from sympy.stats.random_matrix_models import GaussianSymplecticEnsembleModel
  1244. assert _test_args(GaussianSymplecticEnsembleModel('U', 3))
  1245. def test_sympy__stats__random_matrix_models__CircularEnsembleModel():
  1246. from sympy.stats.random_matrix_models import CircularEnsembleModel
  1247. assert _test_args(CircularEnsembleModel('C', 3))
  1248. def test_sympy__stats__random_matrix_models__CircularUnitaryEnsembleModel():
  1249. from sympy.stats.random_matrix_models import CircularUnitaryEnsembleModel
  1250. assert _test_args(CircularUnitaryEnsembleModel('U', 3))
  1251. def test_sympy__stats__random_matrix_models__CircularOrthogonalEnsembleModel():
  1252. from sympy.stats.random_matrix_models import CircularOrthogonalEnsembleModel
  1253. assert _test_args(CircularOrthogonalEnsembleModel('O', 3))
  1254. def test_sympy__stats__random_matrix_models__CircularSymplecticEnsembleModel():
  1255. from sympy.stats.random_matrix_models import CircularSymplecticEnsembleModel
  1256. assert _test_args(CircularSymplecticEnsembleModel('S', 3))
  1257. def test_sympy__stats__symbolic_multivariate_probability__ExpectationMatrix():
  1258. from sympy.stats import ExpectationMatrix
  1259. from sympy.stats.rv import RandomMatrixSymbol
  1260. assert _test_args(ExpectationMatrix(RandomMatrixSymbol('R', 2, 1)))
  1261. def test_sympy__stats__symbolic_multivariate_probability__VarianceMatrix():
  1262. from sympy.stats import VarianceMatrix
  1263. from sympy.stats.rv import RandomMatrixSymbol
  1264. assert _test_args(VarianceMatrix(RandomMatrixSymbol('R', 3, 1)))
  1265. def test_sympy__stats__symbolic_multivariate_probability__CrossCovarianceMatrix():
  1266. from sympy.stats import CrossCovarianceMatrix
  1267. from sympy.stats.rv import RandomMatrixSymbol
  1268. assert _test_args(CrossCovarianceMatrix(RandomMatrixSymbol('R', 3, 1),
  1269. RandomMatrixSymbol('X', 3, 1)))
  1270. def test_sympy__stats__matrix_distributions__MatrixPSpace():
  1271. from sympy.stats.matrix_distributions import MatrixDistribution, MatrixPSpace
  1272. from sympy.matrices.dense import Matrix
  1273. M = MatrixDistribution(1, Matrix([[1, 0], [0, 1]]))
  1274. assert _test_args(MatrixPSpace('M', M, 2, 2))
  1275. def test_sympy__stats__matrix_distributions__MatrixDistribution():
  1276. from sympy.stats.matrix_distributions import MatrixDistribution
  1277. from sympy.matrices.dense import Matrix
  1278. assert _test_args(MatrixDistribution(1, Matrix([[1, 0], [0, 1]])))
  1279. def test_sympy__stats__matrix_distributions__MatrixGammaDistribution():
  1280. from sympy.stats.matrix_distributions import MatrixGammaDistribution
  1281. from sympy.matrices.dense import Matrix
  1282. assert _test_args(MatrixGammaDistribution(3, 4, Matrix([[1, 0], [0, 1]])))
  1283. def test_sympy__stats__matrix_distributions__WishartDistribution():
  1284. from sympy.stats.matrix_distributions import WishartDistribution
  1285. from sympy.matrices.dense import Matrix
  1286. assert _test_args(WishartDistribution(3, Matrix([[1, 0], [0, 1]])))
  1287. def test_sympy__stats__matrix_distributions__MatrixNormalDistribution():
  1288. from sympy.stats.matrix_distributions import MatrixNormalDistribution
  1289. from sympy.matrices.expressions.matexpr import MatrixSymbol
  1290. L = MatrixSymbol('L', 1, 2)
  1291. S1 = MatrixSymbol('S1', 1, 1)
  1292. S2 = MatrixSymbol('S2', 2, 2)
  1293. assert _test_args(MatrixNormalDistribution(L, S1, S2))
  1294. def test_sympy__stats__matrix_distributions__MatrixStudentTDistribution():
  1295. from sympy.stats.matrix_distributions import MatrixStudentTDistribution
  1296. from sympy.matrices.expressions.matexpr import MatrixSymbol
  1297. v = symbols('v', positive=True)
  1298. Omega = MatrixSymbol('Omega', 3, 3)
  1299. Sigma = MatrixSymbol('Sigma', 1, 1)
  1300. Location = MatrixSymbol('Location', 1, 3)
  1301. assert _test_args(MatrixStudentTDistribution(v, Location, Omega, Sigma))
  1302. def test_sympy__utilities__matchpy_connector__WildDot():
  1303. from sympy.utilities.matchpy_connector import WildDot
  1304. assert _test_args(WildDot("w_"))
  1305. def test_sympy__utilities__matchpy_connector__WildPlus():
  1306. from sympy.utilities.matchpy_connector import WildPlus
  1307. assert _test_args(WildPlus("w__"))
  1308. def test_sympy__utilities__matchpy_connector__WildStar():
  1309. from sympy.utilities.matchpy_connector import WildStar
  1310. assert _test_args(WildStar("w___"))
  1311. def test_sympy__core__symbol__Str():
  1312. from sympy.core.symbol import Str
  1313. assert _test_args(Str('t'))
  1314. def test_sympy__core__symbol__Dummy():
  1315. from sympy.core.symbol import Dummy
  1316. assert _test_args(Dummy('t'))
  1317. def test_sympy__core__symbol__Symbol():
  1318. from sympy.core.symbol import Symbol
  1319. assert _test_args(Symbol('t'))
  1320. def test_sympy__core__symbol__Wild():
  1321. from sympy.core.symbol import Wild
  1322. assert _test_args(Wild('x', exclude=[x]))
  1323. @SKIP("abstract class")
  1324. def test_sympy__functions__combinatorial__factorials__CombinatorialFunction():
  1325. pass
  1326. def test_sympy__functions__combinatorial__factorials__FallingFactorial():
  1327. from sympy.functions.combinatorial.factorials import FallingFactorial
  1328. assert _test_args(FallingFactorial(2, x))
  1329. def test_sympy__functions__combinatorial__factorials__MultiFactorial():
  1330. from sympy.functions.combinatorial.factorials import MultiFactorial
  1331. assert _test_args(MultiFactorial(x))
  1332. def test_sympy__functions__combinatorial__factorials__RisingFactorial():
  1333. from sympy.functions.combinatorial.factorials import RisingFactorial
  1334. assert _test_args(RisingFactorial(2, x))
  1335. def test_sympy__functions__combinatorial__factorials__binomial():
  1336. from sympy.functions.combinatorial.factorials import binomial
  1337. assert _test_args(binomial(2, x))
  1338. def test_sympy__functions__combinatorial__factorials__subfactorial():
  1339. from sympy.functions.combinatorial.factorials import subfactorial
  1340. assert _test_args(subfactorial(x))
  1341. def test_sympy__functions__combinatorial__factorials__factorial():
  1342. from sympy.functions.combinatorial.factorials import factorial
  1343. assert _test_args(factorial(x))
  1344. def test_sympy__functions__combinatorial__factorials__factorial2():
  1345. from sympy.functions.combinatorial.factorials import factorial2
  1346. assert _test_args(factorial2(x))
  1347. def test_sympy__functions__combinatorial__numbers__bell():
  1348. from sympy.functions.combinatorial.numbers import bell
  1349. assert _test_args(bell(x, y))
  1350. def test_sympy__functions__combinatorial__numbers__bernoulli():
  1351. from sympy.functions.combinatorial.numbers import bernoulli
  1352. assert _test_args(bernoulli(x))
  1353. def test_sympy__functions__combinatorial__numbers__catalan():
  1354. from sympy.functions.combinatorial.numbers import catalan
  1355. assert _test_args(catalan(x))
  1356. def test_sympy__functions__combinatorial__numbers__genocchi():
  1357. from sympy.functions.combinatorial.numbers import genocchi
  1358. assert _test_args(genocchi(x))
  1359. def test_sympy__functions__combinatorial__numbers__euler():
  1360. from sympy.functions.combinatorial.numbers import euler
  1361. assert _test_args(euler(x))
  1362. def test_sympy__functions__combinatorial__numbers__andre():
  1363. from sympy.functions.combinatorial.numbers import andre
  1364. assert _test_args(andre(x))
  1365. def test_sympy__functions__combinatorial__numbers__carmichael():
  1366. from sympy.functions.combinatorial.numbers import carmichael
  1367. assert _test_args(carmichael(x))
  1368. def test_sympy__functions__combinatorial__numbers__motzkin():
  1369. from sympy.functions.combinatorial.numbers import motzkin
  1370. assert _test_args(motzkin(5))
  1371. def test_sympy__functions__combinatorial__numbers__fibonacci():
  1372. from sympy.functions.combinatorial.numbers import fibonacci
  1373. assert _test_args(fibonacci(x))
  1374. def test_sympy__functions__combinatorial__numbers__tribonacci():
  1375. from sympy.functions.combinatorial.numbers import tribonacci
  1376. assert _test_args(tribonacci(x))
  1377. def test_sympy__functions__combinatorial__numbers__harmonic():
  1378. from sympy.functions.combinatorial.numbers import harmonic
  1379. assert _test_args(harmonic(x, 2))
  1380. def test_sympy__functions__combinatorial__numbers__lucas():
  1381. from sympy.functions.combinatorial.numbers import lucas
  1382. assert _test_args(lucas(x))
  1383. def test_sympy__functions__combinatorial__numbers__partition():
  1384. from sympy.core.symbol import Symbol
  1385. from sympy.functions.combinatorial.numbers import partition
  1386. assert _test_args(partition(Symbol('a', integer=True)))
  1387. def test_sympy__functions__elementary__complexes__Abs():
  1388. from sympy.functions.elementary.complexes import Abs
  1389. assert _test_args(Abs(x))
  1390. def test_sympy__functions__elementary__complexes__adjoint():
  1391. from sympy.functions.elementary.complexes import adjoint
  1392. assert _test_args(adjoint(x))
  1393. def test_sympy__functions__elementary__complexes__arg():
  1394. from sympy.functions.elementary.complexes import arg
  1395. assert _test_args(arg(x))
  1396. def test_sympy__functions__elementary__complexes__conjugate():
  1397. from sympy.functions.elementary.complexes import conjugate
  1398. assert _test_args(conjugate(x))
  1399. def test_sympy__functions__elementary__complexes__im():
  1400. from sympy.functions.elementary.complexes import im
  1401. assert _test_args(im(x))
  1402. def test_sympy__functions__elementary__complexes__re():
  1403. from sympy.functions.elementary.complexes import re
  1404. assert _test_args(re(x))
  1405. def test_sympy__functions__elementary__complexes__sign():
  1406. from sympy.functions.elementary.complexes import sign
  1407. assert _test_args(sign(x))
  1408. def test_sympy__functions__elementary__complexes__polar_lift():
  1409. from sympy.functions.elementary.complexes import polar_lift
  1410. assert _test_args(polar_lift(x))
  1411. def test_sympy__functions__elementary__complexes__periodic_argument():
  1412. from sympy.functions.elementary.complexes import periodic_argument
  1413. assert _test_args(periodic_argument(x, y))
  1414. def test_sympy__functions__elementary__complexes__principal_branch():
  1415. from sympy.functions.elementary.complexes import principal_branch
  1416. assert _test_args(principal_branch(x, y))
  1417. def test_sympy__functions__elementary__complexes__transpose():
  1418. from sympy.functions.elementary.complexes import transpose
  1419. assert _test_args(transpose(x))
  1420. def test_sympy__functions__elementary__exponential__LambertW():
  1421. from sympy.functions.elementary.exponential import LambertW
  1422. assert _test_args(LambertW(2))
  1423. @SKIP("abstract class")
  1424. def test_sympy__functions__elementary__exponential__ExpBase():
  1425. pass
  1426. def test_sympy__functions__elementary__exponential__exp():
  1427. from sympy.functions.elementary.exponential import exp
  1428. assert _test_args(exp(2))
  1429. def test_sympy__functions__elementary__exponential__exp_polar():
  1430. from sympy.functions.elementary.exponential import exp_polar
  1431. assert _test_args(exp_polar(2))
  1432. def test_sympy__functions__elementary__exponential__log():
  1433. from sympy.functions.elementary.exponential import log
  1434. assert _test_args(log(2))
  1435. @SKIP("abstract class")
  1436. def test_sympy__functions__elementary__hyperbolic__HyperbolicFunction():
  1437. pass
  1438. @SKIP("abstract class")
  1439. def test_sympy__functions__elementary__hyperbolic__ReciprocalHyperbolicFunction():
  1440. pass
  1441. @SKIP("abstract class")
  1442. def test_sympy__functions__elementary__hyperbolic__InverseHyperbolicFunction():
  1443. pass
  1444. def test_sympy__functions__elementary__hyperbolic__acosh():
  1445. from sympy.functions.elementary.hyperbolic import acosh
  1446. assert _test_args(acosh(2))
  1447. def test_sympy__functions__elementary__hyperbolic__acoth():
  1448. from sympy.functions.elementary.hyperbolic import acoth
  1449. assert _test_args(acoth(2))
  1450. def test_sympy__functions__elementary__hyperbolic__asinh():
  1451. from sympy.functions.elementary.hyperbolic import asinh
  1452. assert _test_args(asinh(2))
  1453. def test_sympy__functions__elementary__hyperbolic__atanh():
  1454. from sympy.functions.elementary.hyperbolic import atanh
  1455. assert _test_args(atanh(2))
  1456. def test_sympy__functions__elementary__hyperbolic__asech():
  1457. from sympy.functions.elementary.hyperbolic import asech
  1458. assert _test_args(asech(x))
  1459. def test_sympy__functions__elementary__hyperbolic__acsch():
  1460. from sympy.functions.elementary.hyperbolic import acsch
  1461. assert _test_args(acsch(x))
  1462. def test_sympy__functions__elementary__hyperbolic__cosh():
  1463. from sympy.functions.elementary.hyperbolic import cosh
  1464. assert _test_args(cosh(2))
  1465. def test_sympy__functions__elementary__hyperbolic__coth():
  1466. from sympy.functions.elementary.hyperbolic import coth
  1467. assert _test_args(coth(2))
  1468. def test_sympy__functions__elementary__hyperbolic__csch():
  1469. from sympy.functions.elementary.hyperbolic import csch
  1470. assert _test_args(csch(2))
  1471. def test_sympy__functions__elementary__hyperbolic__sech():
  1472. from sympy.functions.elementary.hyperbolic import sech
  1473. assert _test_args(sech(2))
  1474. def test_sympy__functions__elementary__hyperbolic__sinh():
  1475. from sympy.functions.elementary.hyperbolic import sinh
  1476. assert _test_args(sinh(2))
  1477. def test_sympy__functions__elementary__hyperbolic__tanh():
  1478. from sympy.functions.elementary.hyperbolic import tanh
  1479. assert _test_args(tanh(2))
  1480. @SKIP("abstract class")
  1481. def test_sympy__functions__elementary__integers__RoundFunction():
  1482. pass
  1483. def test_sympy__functions__elementary__integers__ceiling():
  1484. from sympy.functions.elementary.integers import ceiling
  1485. assert _test_args(ceiling(x))
  1486. def test_sympy__functions__elementary__integers__floor():
  1487. from sympy.functions.elementary.integers import floor
  1488. assert _test_args(floor(x))
  1489. def test_sympy__functions__elementary__integers__frac():
  1490. from sympy.functions.elementary.integers import frac
  1491. assert _test_args(frac(x))
  1492. def test_sympy__functions__elementary__miscellaneous__IdentityFunction():
  1493. from sympy.functions.elementary.miscellaneous import IdentityFunction
  1494. assert _test_args(IdentityFunction())
  1495. def test_sympy__functions__elementary__miscellaneous__Max():
  1496. from sympy.functions.elementary.miscellaneous import Max
  1497. assert _test_args(Max(x, 2))
  1498. def test_sympy__functions__elementary__miscellaneous__Min():
  1499. from sympy.functions.elementary.miscellaneous import Min
  1500. assert _test_args(Min(x, 2))
  1501. @SKIP("abstract class")
  1502. def test_sympy__functions__elementary__miscellaneous__MinMaxBase():
  1503. pass
  1504. def test_sympy__functions__elementary__miscellaneous__Rem():
  1505. from sympy.functions.elementary.miscellaneous import Rem
  1506. assert _test_args(Rem(x, 2))
  1507. def test_sympy__functions__elementary__piecewise__ExprCondPair():
  1508. from sympy.functions.elementary.piecewise import ExprCondPair
  1509. assert _test_args(ExprCondPair(1, True))
  1510. def test_sympy__functions__elementary__piecewise__Piecewise():
  1511. from sympy.functions.elementary.piecewise import Piecewise
  1512. assert _test_args(Piecewise((1, x >= 0), (0, True)))
  1513. @SKIP("abstract class")
  1514. def test_sympy__functions__elementary__trigonometric__TrigonometricFunction():
  1515. pass
  1516. @SKIP("abstract class")
  1517. def test_sympy__functions__elementary__trigonometric__ReciprocalTrigonometricFunction():
  1518. pass
  1519. @SKIP("abstract class")
  1520. def test_sympy__functions__elementary__trigonometric__InverseTrigonometricFunction():
  1521. pass
  1522. def test_sympy__functions__elementary__trigonometric__acos():
  1523. from sympy.functions.elementary.trigonometric import acos
  1524. assert _test_args(acos(2))
  1525. def test_sympy__functions__elementary__trigonometric__acot():
  1526. from sympy.functions.elementary.trigonometric import acot
  1527. assert _test_args(acot(2))
  1528. def test_sympy__functions__elementary__trigonometric__asin():
  1529. from sympy.functions.elementary.trigonometric import asin
  1530. assert _test_args(asin(2))
  1531. def test_sympy__functions__elementary__trigonometric__asec():
  1532. from sympy.functions.elementary.trigonometric import asec
  1533. assert _test_args(asec(x))
  1534. def test_sympy__functions__elementary__trigonometric__acsc():
  1535. from sympy.functions.elementary.trigonometric import acsc
  1536. assert _test_args(acsc(x))
  1537. def test_sympy__functions__elementary__trigonometric__atan():
  1538. from sympy.functions.elementary.trigonometric import atan
  1539. assert _test_args(atan(2))
  1540. def test_sympy__functions__elementary__trigonometric__atan2():
  1541. from sympy.functions.elementary.trigonometric import atan2
  1542. assert _test_args(atan2(2, 3))
  1543. def test_sympy__functions__elementary__trigonometric__cos():
  1544. from sympy.functions.elementary.trigonometric import cos
  1545. assert _test_args(cos(2))
  1546. def test_sympy__functions__elementary__trigonometric__csc():
  1547. from sympy.functions.elementary.trigonometric import csc
  1548. assert _test_args(csc(2))
  1549. def test_sympy__functions__elementary__trigonometric__cot():
  1550. from sympy.functions.elementary.trigonometric import cot
  1551. assert _test_args(cot(2))
  1552. def test_sympy__functions__elementary__trigonometric__sin():
  1553. assert _test_args(sin(2))
  1554. def test_sympy__functions__elementary__trigonometric__sinc():
  1555. from sympy.functions.elementary.trigonometric import sinc
  1556. assert _test_args(sinc(2))
  1557. def test_sympy__functions__elementary__trigonometric__sec():
  1558. from sympy.functions.elementary.trigonometric import sec
  1559. assert _test_args(sec(2))
  1560. def test_sympy__functions__elementary__trigonometric__tan():
  1561. from sympy.functions.elementary.trigonometric import tan
  1562. assert _test_args(tan(2))
  1563. @SKIP("abstract class")
  1564. def test_sympy__functions__special__bessel__BesselBase():
  1565. pass
  1566. @SKIP("abstract class")
  1567. def test_sympy__functions__special__bessel__SphericalBesselBase():
  1568. pass
  1569. @SKIP("abstract class")
  1570. def test_sympy__functions__special__bessel__SphericalHankelBase():
  1571. pass
  1572. def test_sympy__functions__special__bessel__besseli():
  1573. from sympy.functions.special.bessel import besseli
  1574. assert _test_args(besseli(x, 1))
  1575. def test_sympy__functions__special__bessel__besselj():
  1576. from sympy.functions.special.bessel import besselj
  1577. assert _test_args(besselj(x, 1))
  1578. def test_sympy__functions__special__bessel__besselk():
  1579. from sympy.functions.special.bessel import besselk
  1580. assert _test_args(besselk(x, 1))
  1581. def test_sympy__functions__special__bessel__bessely():
  1582. from sympy.functions.special.bessel import bessely
  1583. assert _test_args(bessely(x, 1))
  1584. def test_sympy__functions__special__bessel__hankel1():
  1585. from sympy.functions.special.bessel import hankel1
  1586. assert _test_args(hankel1(x, 1))
  1587. def test_sympy__functions__special__bessel__hankel2():
  1588. from sympy.functions.special.bessel import hankel2
  1589. assert _test_args(hankel2(x, 1))
  1590. def test_sympy__functions__special__bessel__jn():
  1591. from sympy.functions.special.bessel import jn
  1592. assert _test_args(jn(0, x))
  1593. def test_sympy__functions__special__bessel__yn():
  1594. from sympy.functions.special.bessel import yn
  1595. assert _test_args(yn(0, x))
  1596. def test_sympy__functions__special__bessel__hn1():
  1597. from sympy.functions.special.bessel import hn1
  1598. assert _test_args(hn1(0, x))
  1599. def test_sympy__functions__special__bessel__hn2():
  1600. from sympy.functions.special.bessel import hn2
  1601. assert _test_args(hn2(0, x))
  1602. def test_sympy__functions__special__bessel__AiryBase():
  1603. pass
  1604. def test_sympy__functions__special__bessel__airyai():
  1605. from sympy.functions.special.bessel import airyai
  1606. assert _test_args(airyai(2))
  1607. def test_sympy__functions__special__bessel__airybi():
  1608. from sympy.functions.special.bessel import airybi
  1609. assert _test_args(airybi(2))
  1610. def test_sympy__functions__special__bessel__airyaiprime():
  1611. from sympy.functions.special.bessel import airyaiprime
  1612. assert _test_args(airyaiprime(2))
  1613. def test_sympy__functions__special__bessel__airybiprime():
  1614. from sympy.functions.special.bessel import airybiprime
  1615. assert _test_args(airybiprime(2))
  1616. def test_sympy__functions__special__bessel__marcumq():
  1617. from sympy.functions.special.bessel import marcumq
  1618. assert _test_args(marcumq(x, y, z))
  1619. def test_sympy__functions__special__elliptic_integrals__elliptic_k():
  1620. from sympy.functions.special.elliptic_integrals import elliptic_k as K
  1621. assert _test_args(K(x))
  1622. def test_sympy__functions__special__elliptic_integrals__elliptic_f():
  1623. from sympy.functions.special.elliptic_integrals import elliptic_f as F
  1624. assert _test_args(F(x, y))
  1625. def test_sympy__functions__special__elliptic_integrals__elliptic_e():
  1626. from sympy.functions.special.elliptic_integrals import elliptic_e as E
  1627. assert _test_args(E(x))
  1628. assert _test_args(E(x, y))
  1629. def test_sympy__functions__special__elliptic_integrals__elliptic_pi():
  1630. from sympy.functions.special.elliptic_integrals import elliptic_pi as P
  1631. assert _test_args(P(x, y))
  1632. assert _test_args(P(x, y, z))
  1633. def test_sympy__functions__special__delta_functions__DiracDelta():
  1634. from sympy.functions.special.delta_functions import DiracDelta
  1635. assert _test_args(DiracDelta(x, 1))
  1636. def test_sympy__functions__special__singularity_functions__SingularityFunction():
  1637. from sympy.functions.special.singularity_functions import SingularityFunction
  1638. assert _test_args(SingularityFunction(x, y, z))
  1639. def test_sympy__functions__special__delta_functions__Heaviside():
  1640. from sympy.functions.special.delta_functions import Heaviside
  1641. assert _test_args(Heaviside(x))
  1642. def test_sympy__functions__special__error_functions__erf():
  1643. from sympy.functions.special.error_functions import erf
  1644. assert _test_args(erf(2))
  1645. def test_sympy__functions__special__error_functions__erfc():
  1646. from sympy.functions.special.error_functions import erfc
  1647. assert _test_args(erfc(2))
  1648. def test_sympy__functions__special__error_functions__erfi():
  1649. from sympy.functions.special.error_functions import erfi
  1650. assert _test_args(erfi(2))
  1651. def test_sympy__functions__special__error_functions__erf2():
  1652. from sympy.functions.special.error_functions import erf2
  1653. assert _test_args(erf2(2, 3))
  1654. def test_sympy__functions__special__error_functions__erfinv():
  1655. from sympy.functions.special.error_functions import erfinv
  1656. assert _test_args(erfinv(2))
  1657. def test_sympy__functions__special__error_functions__erfcinv():
  1658. from sympy.functions.special.error_functions import erfcinv
  1659. assert _test_args(erfcinv(2))
  1660. def test_sympy__functions__special__error_functions__erf2inv():
  1661. from sympy.functions.special.error_functions import erf2inv
  1662. assert _test_args(erf2inv(2, 3))
  1663. @SKIP("abstract class")
  1664. def test_sympy__functions__special__error_functions__FresnelIntegral():
  1665. pass
  1666. def test_sympy__functions__special__error_functions__fresnels():
  1667. from sympy.functions.special.error_functions import fresnels
  1668. assert _test_args(fresnels(2))
  1669. def test_sympy__functions__special__error_functions__fresnelc():
  1670. from sympy.functions.special.error_functions import fresnelc
  1671. assert _test_args(fresnelc(2))
  1672. def test_sympy__functions__special__error_functions__erfs():
  1673. from sympy.functions.special.error_functions import _erfs
  1674. assert _test_args(_erfs(2))
  1675. def test_sympy__functions__special__error_functions__Ei():
  1676. from sympy.functions.special.error_functions import Ei
  1677. assert _test_args(Ei(2))
  1678. def test_sympy__functions__special__error_functions__li():
  1679. from sympy.functions.special.error_functions import li
  1680. assert _test_args(li(2))
  1681. def test_sympy__functions__special__error_functions__Li():
  1682. from sympy.functions.special.error_functions import Li
  1683. assert _test_args(Li(5))
  1684. @SKIP("abstract class")
  1685. def test_sympy__functions__special__error_functions__TrigonometricIntegral():
  1686. pass
  1687. def test_sympy__functions__special__error_functions__Si():
  1688. from sympy.functions.special.error_functions import Si
  1689. assert _test_args(Si(2))
  1690. def test_sympy__functions__special__error_functions__Ci():
  1691. from sympy.functions.special.error_functions import Ci
  1692. assert _test_args(Ci(2))
  1693. def test_sympy__functions__special__error_functions__Shi():
  1694. from sympy.functions.special.error_functions import Shi
  1695. assert _test_args(Shi(2))
  1696. def test_sympy__functions__special__error_functions__Chi():
  1697. from sympy.functions.special.error_functions import Chi
  1698. assert _test_args(Chi(2))
  1699. def test_sympy__functions__special__error_functions__expint():
  1700. from sympy.functions.special.error_functions import expint
  1701. assert _test_args(expint(y, x))
  1702. def test_sympy__functions__special__gamma_functions__gamma():
  1703. from sympy.functions.special.gamma_functions import gamma
  1704. assert _test_args(gamma(x))
  1705. def test_sympy__functions__special__gamma_functions__loggamma():
  1706. from sympy.functions.special.gamma_functions import loggamma
  1707. assert _test_args(loggamma(x))
  1708. def test_sympy__functions__special__gamma_functions__lowergamma():
  1709. from sympy.functions.special.gamma_functions import lowergamma
  1710. assert _test_args(lowergamma(x, 2))
  1711. def test_sympy__functions__special__gamma_functions__polygamma():
  1712. from sympy.functions.special.gamma_functions import polygamma
  1713. assert _test_args(polygamma(x, 2))
  1714. def test_sympy__functions__special__gamma_functions__digamma():
  1715. from sympy.functions.special.gamma_functions import digamma
  1716. assert _test_args(digamma(x))
  1717. def test_sympy__functions__special__gamma_functions__trigamma():
  1718. from sympy.functions.special.gamma_functions import trigamma
  1719. assert _test_args(trigamma(x))
  1720. def test_sympy__functions__special__gamma_functions__uppergamma():
  1721. from sympy.functions.special.gamma_functions import uppergamma
  1722. assert _test_args(uppergamma(x, 2))
  1723. def test_sympy__functions__special__gamma_functions__multigamma():
  1724. from sympy.functions.special.gamma_functions import multigamma
  1725. assert _test_args(multigamma(x, 1))
  1726. def test_sympy__functions__special__beta_functions__beta():
  1727. from sympy.functions.special.beta_functions import beta
  1728. assert _test_args(beta(x))
  1729. assert _test_args(beta(x, x))
  1730. def test_sympy__functions__special__beta_functions__betainc():
  1731. from sympy.functions.special.beta_functions import betainc
  1732. assert _test_args(betainc(a, b, x, y))
  1733. def test_sympy__functions__special__beta_functions__betainc_regularized():
  1734. from sympy.functions.special.beta_functions import betainc_regularized
  1735. assert _test_args(betainc_regularized(a, b, x, y))
  1736. def test_sympy__functions__special__mathieu_functions__MathieuBase():
  1737. pass
  1738. def test_sympy__functions__special__mathieu_functions__mathieus():
  1739. from sympy.functions.special.mathieu_functions import mathieus
  1740. assert _test_args(mathieus(1, 1, 1))
  1741. def test_sympy__functions__special__mathieu_functions__mathieuc():
  1742. from sympy.functions.special.mathieu_functions import mathieuc
  1743. assert _test_args(mathieuc(1, 1, 1))
  1744. def test_sympy__functions__special__mathieu_functions__mathieusprime():
  1745. from sympy.functions.special.mathieu_functions import mathieusprime
  1746. assert _test_args(mathieusprime(1, 1, 1))
  1747. def test_sympy__functions__special__mathieu_functions__mathieucprime():
  1748. from sympy.functions.special.mathieu_functions import mathieucprime
  1749. assert _test_args(mathieucprime(1, 1, 1))
  1750. @SKIP("abstract class")
  1751. def test_sympy__functions__special__hyper__TupleParametersBase():
  1752. pass
  1753. @SKIP("abstract class")
  1754. def test_sympy__functions__special__hyper__TupleArg():
  1755. pass
  1756. def test_sympy__functions__special__hyper__hyper():
  1757. from sympy.functions.special.hyper import hyper
  1758. assert _test_args(hyper([1, 2, 3], [4, 5], x))
  1759. def test_sympy__functions__special__hyper__meijerg():
  1760. from sympy.functions.special.hyper import meijerg
  1761. assert _test_args(meijerg([1, 2, 3], [4, 5], [6], [], x))
  1762. @SKIP("abstract class")
  1763. def test_sympy__functions__special__hyper__HyperRep():
  1764. pass
  1765. def test_sympy__functions__special__hyper__HyperRep_power1():
  1766. from sympy.functions.special.hyper import HyperRep_power1
  1767. assert _test_args(HyperRep_power1(x, y))
  1768. def test_sympy__functions__special__hyper__HyperRep_power2():
  1769. from sympy.functions.special.hyper import HyperRep_power2
  1770. assert _test_args(HyperRep_power2(x, y))
  1771. def test_sympy__functions__special__hyper__HyperRep_log1():
  1772. from sympy.functions.special.hyper import HyperRep_log1
  1773. assert _test_args(HyperRep_log1(x))
  1774. def test_sympy__functions__special__hyper__HyperRep_atanh():
  1775. from sympy.functions.special.hyper import HyperRep_atanh
  1776. assert _test_args(HyperRep_atanh(x))
  1777. def test_sympy__functions__special__hyper__HyperRep_asin1():
  1778. from sympy.functions.special.hyper import HyperRep_asin1
  1779. assert _test_args(HyperRep_asin1(x))
  1780. def test_sympy__functions__special__hyper__HyperRep_asin2():
  1781. from sympy.functions.special.hyper import HyperRep_asin2
  1782. assert _test_args(HyperRep_asin2(x))
  1783. def test_sympy__functions__special__hyper__HyperRep_sqrts1():
  1784. from sympy.functions.special.hyper import HyperRep_sqrts1
  1785. assert _test_args(HyperRep_sqrts1(x, y))
  1786. def test_sympy__functions__special__hyper__HyperRep_sqrts2():
  1787. from sympy.functions.special.hyper import HyperRep_sqrts2
  1788. assert _test_args(HyperRep_sqrts2(x, y))
  1789. def test_sympy__functions__special__hyper__HyperRep_log2():
  1790. from sympy.functions.special.hyper import HyperRep_log2
  1791. assert _test_args(HyperRep_log2(x))
  1792. def test_sympy__functions__special__hyper__HyperRep_cosasin():
  1793. from sympy.functions.special.hyper import HyperRep_cosasin
  1794. assert _test_args(HyperRep_cosasin(x, y))
  1795. def test_sympy__functions__special__hyper__HyperRep_sinasin():
  1796. from sympy.functions.special.hyper import HyperRep_sinasin
  1797. assert _test_args(HyperRep_sinasin(x, y))
  1798. def test_sympy__functions__special__hyper__appellf1():
  1799. from sympy.functions.special.hyper import appellf1
  1800. a, b1, b2, c, x, y = symbols('a b1 b2 c x y')
  1801. assert _test_args(appellf1(a, b1, b2, c, x, y))
  1802. @SKIP("abstract class")
  1803. def test_sympy__functions__special__polynomials__OrthogonalPolynomial():
  1804. pass
  1805. def test_sympy__functions__special__polynomials__jacobi():
  1806. from sympy.functions.special.polynomials import jacobi
  1807. assert _test_args(jacobi(x, y, 2, 2))
  1808. def test_sympy__functions__special__polynomials__gegenbauer():
  1809. from sympy.functions.special.polynomials import gegenbauer
  1810. assert _test_args(gegenbauer(x, 2, 2))
  1811. def test_sympy__functions__special__polynomials__chebyshevt():
  1812. from sympy.functions.special.polynomials import chebyshevt
  1813. assert _test_args(chebyshevt(x, 2))
  1814. def test_sympy__functions__special__polynomials__chebyshevt_root():
  1815. from sympy.functions.special.polynomials import chebyshevt_root
  1816. assert _test_args(chebyshevt_root(3, 2))
  1817. def test_sympy__functions__special__polynomials__chebyshevu():
  1818. from sympy.functions.special.polynomials import chebyshevu
  1819. assert _test_args(chebyshevu(x, 2))
  1820. def test_sympy__functions__special__polynomials__chebyshevu_root():
  1821. from sympy.functions.special.polynomials import chebyshevu_root
  1822. assert _test_args(chebyshevu_root(3, 2))
  1823. def test_sympy__functions__special__polynomials__hermite():
  1824. from sympy.functions.special.polynomials import hermite
  1825. assert _test_args(hermite(x, 2))
  1826. def test_sympy__functions__special__polynomials__hermite_prob():
  1827. from sympy.functions.special.polynomials import hermite_prob
  1828. assert _test_args(hermite_prob(x, 2))
  1829. def test_sympy__functions__special__polynomials__legendre():
  1830. from sympy.functions.special.polynomials import legendre
  1831. assert _test_args(legendre(x, 2))
  1832. def test_sympy__functions__special__polynomials__assoc_legendre():
  1833. from sympy.functions.special.polynomials import assoc_legendre
  1834. assert _test_args(assoc_legendre(x, 0, y))
  1835. def test_sympy__functions__special__polynomials__laguerre():
  1836. from sympy.functions.special.polynomials import laguerre
  1837. assert _test_args(laguerre(x, 2))
  1838. def test_sympy__functions__special__polynomials__assoc_laguerre():
  1839. from sympy.functions.special.polynomials import assoc_laguerre
  1840. assert _test_args(assoc_laguerre(x, 0, y))
  1841. def test_sympy__functions__special__spherical_harmonics__Ynm():
  1842. from sympy.functions.special.spherical_harmonics import Ynm
  1843. assert _test_args(Ynm(1, 1, x, y))
  1844. def test_sympy__functions__special__spherical_harmonics__Znm():
  1845. from sympy.functions.special.spherical_harmonics import Znm
  1846. assert _test_args(Znm(x, y, 1, 1))
  1847. def test_sympy__functions__special__tensor_functions__LeviCivita():
  1848. from sympy.functions.special.tensor_functions import LeviCivita
  1849. assert _test_args(LeviCivita(x, y, 2))
  1850. def test_sympy__functions__special__tensor_functions__KroneckerDelta():
  1851. from sympy.functions.special.tensor_functions import KroneckerDelta
  1852. assert _test_args(KroneckerDelta(x, y))
  1853. def test_sympy__functions__special__zeta_functions__dirichlet_eta():
  1854. from sympy.functions.special.zeta_functions import dirichlet_eta
  1855. assert _test_args(dirichlet_eta(x))
  1856. def test_sympy__functions__special__zeta_functions__riemann_xi():
  1857. from sympy.functions.special.zeta_functions import riemann_xi
  1858. assert _test_args(riemann_xi(x))
  1859. def test_sympy__functions__special__zeta_functions__zeta():
  1860. from sympy.functions.special.zeta_functions import zeta
  1861. assert _test_args(zeta(101))
  1862. def test_sympy__functions__special__zeta_functions__lerchphi():
  1863. from sympy.functions.special.zeta_functions import lerchphi
  1864. assert _test_args(lerchphi(x, y, z))
  1865. def test_sympy__functions__special__zeta_functions__polylog():
  1866. from sympy.functions.special.zeta_functions import polylog
  1867. assert _test_args(polylog(x, y))
  1868. def test_sympy__functions__special__zeta_functions__stieltjes():
  1869. from sympy.functions.special.zeta_functions import stieltjes
  1870. assert _test_args(stieltjes(x, y))
  1871. def test_sympy__integrals__integrals__Integral():
  1872. from sympy.integrals.integrals import Integral
  1873. assert _test_args(Integral(2, (x, 0, 1)))
  1874. def test_sympy__integrals__risch__NonElementaryIntegral():
  1875. from sympy.integrals.risch import NonElementaryIntegral
  1876. assert _test_args(NonElementaryIntegral(exp(-x**2), x))
  1877. @SKIP("abstract class")
  1878. def test_sympy__integrals__transforms__IntegralTransform():
  1879. pass
  1880. def test_sympy__integrals__transforms__MellinTransform():
  1881. from sympy.integrals.transforms import MellinTransform
  1882. assert _test_args(MellinTransform(2, x, y))
  1883. def test_sympy__integrals__transforms__InverseMellinTransform():
  1884. from sympy.integrals.transforms import InverseMellinTransform
  1885. assert _test_args(InverseMellinTransform(2, x, y, 0, 1))
  1886. def test_sympy__integrals__laplace__LaplaceTransform():
  1887. from sympy.integrals.laplace import LaplaceTransform
  1888. assert _test_args(LaplaceTransform(2, x, y))
  1889. def test_sympy__integrals__laplace__InverseLaplaceTransform():
  1890. from sympy.integrals.laplace import InverseLaplaceTransform
  1891. assert _test_args(InverseLaplaceTransform(2, x, y, 0))
  1892. @SKIP("abstract class")
  1893. def test_sympy__integrals__transforms__FourierTypeTransform():
  1894. pass
  1895. def test_sympy__integrals__transforms__InverseFourierTransform():
  1896. from sympy.integrals.transforms import InverseFourierTransform
  1897. assert _test_args(InverseFourierTransform(2, x, y))
  1898. def test_sympy__integrals__transforms__FourierTransform():
  1899. from sympy.integrals.transforms import FourierTransform
  1900. assert _test_args(FourierTransform(2, x, y))
  1901. @SKIP("abstract class")
  1902. def test_sympy__integrals__transforms__SineCosineTypeTransform():
  1903. pass
  1904. def test_sympy__integrals__transforms__InverseSineTransform():
  1905. from sympy.integrals.transforms import InverseSineTransform
  1906. assert _test_args(InverseSineTransform(2, x, y))
  1907. def test_sympy__integrals__transforms__SineTransform():
  1908. from sympy.integrals.transforms import SineTransform
  1909. assert _test_args(SineTransform(2, x, y))
  1910. def test_sympy__integrals__transforms__InverseCosineTransform():
  1911. from sympy.integrals.transforms import InverseCosineTransform
  1912. assert _test_args(InverseCosineTransform(2, x, y))
  1913. def test_sympy__integrals__transforms__CosineTransform():
  1914. from sympy.integrals.transforms import CosineTransform
  1915. assert _test_args(CosineTransform(2, x, y))
  1916. @SKIP("abstract class")
  1917. def test_sympy__integrals__transforms__HankelTypeTransform():
  1918. pass
  1919. def test_sympy__integrals__transforms__InverseHankelTransform():
  1920. from sympy.integrals.transforms import InverseHankelTransform
  1921. assert _test_args(InverseHankelTransform(2, x, y, 0))
  1922. def test_sympy__integrals__transforms__HankelTransform():
  1923. from sympy.integrals.transforms import HankelTransform
  1924. assert _test_args(HankelTransform(2, x, y, 0))
  1925. def test_sympy__liealgebras__cartan_type__Standard_Cartan():
  1926. from sympy.liealgebras.cartan_type import Standard_Cartan
  1927. assert _test_args(Standard_Cartan("A", 2))
  1928. def test_sympy__liealgebras__weyl_group__WeylGroup():
  1929. from sympy.liealgebras.weyl_group import WeylGroup
  1930. assert _test_args(WeylGroup("B4"))
  1931. def test_sympy__liealgebras__root_system__RootSystem():
  1932. from sympy.liealgebras.root_system import RootSystem
  1933. assert _test_args(RootSystem("A2"))
  1934. def test_sympy__liealgebras__type_a__TypeA():
  1935. from sympy.liealgebras.type_a import TypeA
  1936. assert _test_args(TypeA(2))
  1937. def test_sympy__liealgebras__type_b__TypeB():
  1938. from sympy.liealgebras.type_b import TypeB
  1939. assert _test_args(TypeB(4))
  1940. def test_sympy__liealgebras__type_c__TypeC():
  1941. from sympy.liealgebras.type_c import TypeC
  1942. assert _test_args(TypeC(4))
  1943. def test_sympy__liealgebras__type_d__TypeD():
  1944. from sympy.liealgebras.type_d import TypeD
  1945. assert _test_args(TypeD(4))
  1946. def test_sympy__liealgebras__type_e__TypeE():
  1947. from sympy.liealgebras.type_e import TypeE
  1948. assert _test_args(TypeE(6))
  1949. def test_sympy__liealgebras__type_f__TypeF():
  1950. from sympy.liealgebras.type_f import TypeF
  1951. assert _test_args(TypeF(4))
  1952. def test_sympy__liealgebras__type_g__TypeG():
  1953. from sympy.liealgebras.type_g import TypeG
  1954. assert _test_args(TypeG(2))
  1955. def test_sympy__logic__boolalg__And():
  1956. from sympy.logic.boolalg import And
  1957. assert _test_args(And(x, y, 1))
  1958. @SKIP("abstract class")
  1959. def test_sympy__logic__boolalg__Boolean():
  1960. pass
  1961. def test_sympy__logic__boolalg__BooleanFunction():
  1962. from sympy.logic.boolalg import BooleanFunction
  1963. assert _test_args(BooleanFunction(1, 2, 3))
  1964. @SKIP("abstract class")
  1965. def test_sympy__logic__boolalg__BooleanAtom():
  1966. pass
  1967. def test_sympy__logic__boolalg__BooleanTrue():
  1968. from sympy.logic.boolalg import true
  1969. assert _test_args(true)
  1970. def test_sympy__logic__boolalg__BooleanFalse():
  1971. from sympy.logic.boolalg import false
  1972. assert _test_args(false)
  1973. def test_sympy__logic__boolalg__Equivalent():
  1974. from sympy.logic.boolalg import Equivalent
  1975. assert _test_args(Equivalent(x, 2))
  1976. def test_sympy__logic__boolalg__ITE():
  1977. from sympy.logic.boolalg import ITE
  1978. assert _test_args(ITE(x, y, 1))
  1979. def test_sympy__logic__boolalg__Implies():
  1980. from sympy.logic.boolalg import Implies
  1981. assert _test_args(Implies(x, y))
  1982. def test_sympy__logic__boolalg__Nand():
  1983. from sympy.logic.boolalg import Nand
  1984. assert _test_args(Nand(x, y, 1))
  1985. def test_sympy__logic__boolalg__Nor():
  1986. from sympy.logic.boolalg import Nor
  1987. assert _test_args(Nor(x, y))
  1988. def test_sympy__logic__boolalg__Not():
  1989. from sympy.logic.boolalg import Not
  1990. assert _test_args(Not(x))
  1991. def test_sympy__logic__boolalg__Or():
  1992. from sympy.logic.boolalg import Or
  1993. assert _test_args(Or(x, y))
  1994. def test_sympy__logic__boolalg__Xor():
  1995. from sympy.logic.boolalg import Xor
  1996. assert _test_args(Xor(x, y, 2))
  1997. def test_sympy__logic__boolalg__Xnor():
  1998. from sympy.logic.boolalg import Xnor
  1999. assert _test_args(Xnor(x, y, 2))
  2000. def test_sympy__logic__boolalg__Exclusive():
  2001. from sympy.logic.boolalg import Exclusive
  2002. assert _test_args(Exclusive(x, y, z))
  2003. def test_sympy__matrices__matrices__DeferredVector():
  2004. from sympy.matrices.matrices import DeferredVector
  2005. assert _test_args(DeferredVector("X"))
  2006. @SKIP("abstract class")
  2007. def test_sympy__matrices__expressions__matexpr__MatrixBase():
  2008. pass
  2009. @SKIP("abstract class")
  2010. def test_sympy__matrices__immutable__ImmutableRepMatrix():
  2011. pass
  2012. def test_sympy__matrices__immutable__ImmutableDenseMatrix():
  2013. from sympy.matrices.immutable import ImmutableDenseMatrix
  2014. m = ImmutableDenseMatrix([[1, 2], [3, 4]])
  2015. assert _test_args(m)
  2016. assert _test_args(Basic(*list(m)))
  2017. m = ImmutableDenseMatrix(1, 1, [1])
  2018. assert _test_args(m)
  2019. assert _test_args(Basic(*list(m)))
  2020. m = ImmutableDenseMatrix(2, 2, lambda i, j: 1)
  2021. assert m[0, 0] is S.One
  2022. m = ImmutableDenseMatrix(2, 2, lambda i, j: 1/(1 + i) + 1/(1 + j))
  2023. assert m[1, 1] is S.One # true div. will give 1.0 if i,j not sympified
  2024. assert _test_args(m)
  2025. assert _test_args(Basic(*list(m)))
  2026. def test_sympy__matrices__immutable__ImmutableSparseMatrix():
  2027. from sympy.matrices.immutable import ImmutableSparseMatrix
  2028. m = ImmutableSparseMatrix([[1, 2], [3, 4]])
  2029. assert _test_args(m)
  2030. assert _test_args(Basic(*list(m)))
  2031. m = ImmutableSparseMatrix(1, 1, {(0, 0): 1})
  2032. assert _test_args(m)
  2033. assert _test_args(Basic(*list(m)))
  2034. m = ImmutableSparseMatrix(1, 1, [1])
  2035. assert _test_args(m)
  2036. assert _test_args(Basic(*list(m)))
  2037. m = ImmutableSparseMatrix(2, 2, lambda i, j: 1)
  2038. assert m[0, 0] is S.One
  2039. m = ImmutableSparseMatrix(2, 2, lambda i, j: 1/(1 + i) + 1/(1 + j))
  2040. assert m[1, 1] is S.One # true div. will give 1.0 if i,j not sympified
  2041. assert _test_args(m)
  2042. assert _test_args(Basic(*list(m)))
  2043. def test_sympy__matrices__expressions__slice__MatrixSlice():
  2044. from sympy.matrices.expressions.slice import MatrixSlice
  2045. from sympy.matrices.expressions import MatrixSymbol
  2046. X = MatrixSymbol('X', 4, 4)
  2047. assert _test_args(MatrixSlice(X, (0, 2), (0, 2)))
  2048. def test_sympy__matrices__expressions__applyfunc__ElementwiseApplyFunction():
  2049. from sympy.matrices.expressions.applyfunc import ElementwiseApplyFunction
  2050. from sympy.matrices.expressions import MatrixSymbol
  2051. X = MatrixSymbol("X", x, x)
  2052. func = Lambda(x, x**2)
  2053. assert _test_args(ElementwiseApplyFunction(func, X))
  2054. def test_sympy__matrices__expressions__blockmatrix__BlockDiagMatrix():
  2055. from sympy.matrices.expressions.blockmatrix import BlockDiagMatrix
  2056. from sympy.matrices.expressions import MatrixSymbol
  2057. X = MatrixSymbol('X', x, x)
  2058. Y = MatrixSymbol('Y', y, y)
  2059. assert _test_args(BlockDiagMatrix(X, Y))
  2060. def test_sympy__matrices__expressions__blockmatrix__BlockMatrix():
  2061. from sympy.matrices.expressions.blockmatrix import BlockMatrix
  2062. from sympy.matrices.expressions import MatrixSymbol, ZeroMatrix
  2063. X = MatrixSymbol('X', x, x)
  2064. Y = MatrixSymbol('Y', y, y)
  2065. Z = MatrixSymbol('Z', x, y)
  2066. O = ZeroMatrix(y, x)
  2067. assert _test_args(BlockMatrix([[X, Z], [O, Y]]))
  2068. def test_sympy__matrices__expressions__inverse__Inverse():
  2069. from sympy.matrices.expressions.inverse import Inverse
  2070. from sympy.matrices.expressions import MatrixSymbol
  2071. assert _test_args(Inverse(MatrixSymbol('A', 3, 3)))
  2072. def test_sympy__matrices__expressions__matadd__MatAdd():
  2073. from sympy.matrices.expressions.matadd import MatAdd
  2074. from sympy.matrices.expressions import MatrixSymbol
  2075. X = MatrixSymbol('X', x, y)
  2076. Y = MatrixSymbol('Y', x, y)
  2077. assert _test_args(MatAdd(X, Y))
  2078. @SKIP("abstract class")
  2079. def test_sympy__matrices__expressions__matexpr__MatrixExpr():
  2080. pass
  2081. def test_sympy__matrices__expressions__matexpr__MatrixElement():
  2082. from sympy.matrices.expressions.matexpr import MatrixSymbol, MatrixElement
  2083. from sympy.core.singleton import S
  2084. assert _test_args(MatrixElement(MatrixSymbol('A', 3, 5), S(2), S(3)))
  2085. def test_sympy__matrices__expressions__matexpr__MatrixSymbol():
  2086. from sympy.matrices.expressions.matexpr import MatrixSymbol
  2087. assert _test_args(MatrixSymbol('A', 3, 5))
  2088. def test_sympy__matrices__expressions__special__OneMatrix():
  2089. from sympy.matrices.expressions.special import OneMatrix
  2090. assert _test_args(OneMatrix(3, 5))
  2091. def test_sympy__matrices__expressions__special__ZeroMatrix():
  2092. from sympy.matrices.expressions.special import ZeroMatrix
  2093. assert _test_args(ZeroMatrix(3, 5))
  2094. def test_sympy__matrices__expressions__special__GenericZeroMatrix():
  2095. from sympy.matrices.expressions.special import GenericZeroMatrix
  2096. assert _test_args(GenericZeroMatrix())
  2097. def test_sympy__matrices__expressions__special__Identity():
  2098. from sympy.matrices.expressions.special import Identity
  2099. assert _test_args(Identity(3))
  2100. def test_sympy__matrices__expressions__special__GenericIdentity():
  2101. from sympy.matrices.expressions.special import GenericIdentity
  2102. assert _test_args(GenericIdentity())
  2103. def test_sympy__matrices__expressions__sets__MatrixSet():
  2104. from sympy.matrices.expressions.sets import MatrixSet
  2105. from sympy.core.singleton import S
  2106. assert _test_args(MatrixSet(2, 2, S.Reals))
  2107. def test_sympy__matrices__expressions__matmul__MatMul():
  2108. from sympy.matrices.expressions.matmul import MatMul
  2109. from sympy.matrices.expressions import MatrixSymbol
  2110. X = MatrixSymbol('X', x, y)
  2111. Y = MatrixSymbol('Y', y, x)
  2112. assert _test_args(MatMul(X, Y))
  2113. def test_sympy__matrices__expressions__dotproduct__DotProduct():
  2114. from sympy.matrices.expressions.dotproduct import DotProduct
  2115. from sympy.matrices.expressions import MatrixSymbol
  2116. X = MatrixSymbol('X', x, 1)
  2117. Y = MatrixSymbol('Y', x, 1)
  2118. assert _test_args(DotProduct(X, Y))
  2119. def test_sympy__matrices__expressions__diagonal__DiagonalMatrix():
  2120. from sympy.matrices.expressions.diagonal import DiagonalMatrix
  2121. from sympy.matrices.expressions import MatrixSymbol
  2122. x = MatrixSymbol('x', 10, 1)
  2123. assert _test_args(DiagonalMatrix(x))
  2124. def test_sympy__matrices__expressions__diagonal__DiagonalOf():
  2125. from sympy.matrices.expressions.diagonal import DiagonalOf
  2126. from sympy.matrices.expressions import MatrixSymbol
  2127. X = MatrixSymbol('x', 10, 10)
  2128. assert _test_args(DiagonalOf(X))
  2129. def test_sympy__matrices__expressions__diagonal__DiagMatrix():
  2130. from sympy.matrices.expressions.diagonal import DiagMatrix
  2131. from sympy.matrices.expressions import MatrixSymbol
  2132. x = MatrixSymbol('x', 10, 1)
  2133. assert _test_args(DiagMatrix(x))
  2134. def test_sympy__matrices__expressions__hadamard__HadamardProduct():
  2135. from sympy.matrices.expressions.hadamard import HadamardProduct
  2136. from sympy.matrices.expressions import MatrixSymbol
  2137. X = MatrixSymbol('X', x, y)
  2138. Y = MatrixSymbol('Y', x, y)
  2139. assert _test_args(HadamardProduct(X, Y))
  2140. def test_sympy__matrices__expressions__hadamard__HadamardPower():
  2141. from sympy.matrices.expressions.hadamard import HadamardPower
  2142. from sympy.matrices.expressions import MatrixSymbol
  2143. from sympy.core.symbol import Symbol
  2144. X = MatrixSymbol('X', x, y)
  2145. n = Symbol("n")
  2146. assert _test_args(HadamardPower(X, n))
  2147. def test_sympy__matrices__expressions__kronecker__KroneckerProduct():
  2148. from sympy.matrices.expressions.kronecker import KroneckerProduct
  2149. from sympy.matrices.expressions import MatrixSymbol
  2150. X = MatrixSymbol('X', x, y)
  2151. Y = MatrixSymbol('Y', x, y)
  2152. assert _test_args(KroneckerProduct(X, Y))
  2153. def test_sympy__matrices__expressions__matpow__MatPow():
  2154. from sympy.matrices.expressions.matpow import MatPow
  2155. from sympy.matrices.expressions import MatrixSymbol
  2156. X = MatrixSymbol('X', x, x)
  2157. assert _test_args(MatPow(X, 2))
  2158. def test_sympy__matrices__expressions__transpose__Transpose():
  2159. from sympy.matrices.expressions.transpose import Transpose
  2160. from sympy.matrices.expressions import MatrixSymbol
  2161. assert _test_args(Transpose(MatrixSymbol('A', 3, 5)))
  2162. def test_sympy__matrices__expressions__adjoint__Adjoint():
  2163. from sympy.matrices.expressions.adjoint import Adjoint
  2164. from sympy.matrices.expressions import MatrixSymbol
  2165. assert _test_args(Adjoint(MatrixSymbol('A', 3, 5)))
  2166. def test_sympy__matrices__expressions__trace__Trace():
  2167. from sympy.matrices.expressions.trace import Trace
  2168. from sympy.matrices.expressions import MatrixSymbol
  2169. assert _test_args(Trace(MatrixSymbol('A', 3, 3)))
  2170. def test_sympy__matrices__expressions__determinant__Determinant():
  2171. from sympy.matrices.expressions.determinant import Determinant
  2172. from sympy.matrices.expressions import MatrixSymbol
  2173. assert _test_args(Determinant(MatrixSymbol('A', 3, 3)))
  2174. def test_sympy__matrices__expressions__determinant__Permanent():
  2175. from sympy.matrices.expressions.determinant import Permanent
  2176. from sympy.matrices.expressions import MatrixSymbol
  2177. assert _test_args(Permanent(MatrixSymbol('A', 3, 4)))
  2178. def test_sympy__matrices__expressions__funcmatrix__FunctionMatrix():
  2179. from sympy.matrices.expressions.funcmatrix import FunctionMatrix
  2180. from sympy.core.symbol import symbols
  2181. i, j = symbols('i,j')
  2182. assert _test_args(FunctionMatrix(3, 3, Lambda((i, j), i - j) ))
  2183. def test_sympy__matrices__expressions__fourier__DFT():
  2184. from sympy.matrices.expressions.fourier import DFT
  2185. from sympy.core.singleton import S
  2186. assert _test_args(DFT(S(2)))
  2187. def test_sympy__matrices__expressions__fourier__IDFT():
  2188. from sympy.matrices.expressions.fourier import IDFT
  2189. from sympy.core.singleton import S
  2190. assert _test_args(IDFT(S(2)))
  2191. from sympy.matrices.expressions import MatrixSymbol
  2192. X = MatrixSymbol('X', 10, 10)
  2193. def test_sympy__matrices__expressions__factorizations__LofLU():
  2194. from sympy.matrices.expressions.factorizations import LofLU
  2195. assert _test_args(LofLU(X))
  2196. def test_sympy__matrices__expressions__factorizations__UofLU():
  2197. from sympy.matrices.expressions.factorizations import UofLU
  2198. assert _test_args(UofLU(X))
  2199. def test_sympy__matrices__expressions__factorizations__QofQR():
  2200. from sympy.matrices.expressions.factorizations import QofQR
  2201. assert _test_args(QofQR(X))
  2202. def test_sympy__matrices__expressions__factorizations__RofQR():
  2203. from sympy.matrices.expressions.factorizations import RofQR
  2204. assert _test_args(RofQR(X))
  2205. def test_sympy__matrices__expressions__factorizations__LofCholesky():
  2206. from sympy.matrices.expressions.factorizations import LofCholesky
  2207. assert _test_args(LofCholesky(X))
  2208. def test_sympy__matrices__expressions__factorizations__UofCholesky():
  2209. from sympy.matrices.expressions.factorizations import UofCholesky
  2210. assert _test_args(UofCholesky(X))
  2211. def test_sympy__matrices__expressions__factorizations__EigenVectors():
  2212. from sympy.matrices.expressions.factorizations import EigenVectors
  2213. assert _test_args(EigenVectors(X))
  2214. def test_sympy__matrices__expressions__factorizations__EigenValues():
  2215. from sympy.matrices.expressions.factorizations import EigenValues
  2216. assert _test_args(EigenValues(X))
  2217. def test_sympy__matrices__expressions__factorizations__UofSVD():
  2218. from sympy.matrices.expressions.factorizations import UofSVD
  2219. assert _test_args(UofSVD(X))
  2220. def test_sympy__matrices__expressions__factorizations__VofSVD():
  2221. from sympy.matrices.expressions.factorizations import VofSVD
  2222. assert _test_args(VofSVD(X))
  2223. def test_sympy__matrices__expressions__factorizations__SofSVD():
  2224. from sympy.matrices.expressions.factorizations import SofSVD
  2225. assert _test_args(SofSVD(X))
  2226. @SKIP("abstract class")
  2227. def test_sympy__matrices__expressions__factorizations__Factorization():
  2228. pass
  2229. def test_sympy__matrices__expressions__permutation__PermutationMatrix():
  2230. from sympy.combinatorics import Permutation
  2231. from sympy.matrices.expressions.permutation import PermutationMatrix
  2232. assert _test_args(PermutationMatrix(Permutation([2, 0, 1])))
  2233. def test_sympy__matrices__expressions__permutation__MatrixPermute():
  2234. from sympy.combinatorics import Permutation
  2235. from sympy.matrices.expressions.matexpr import MatrixSymbol
  2236. from sympy.matrices.expressions.permutation import MatrixPermute
  2237. A = MatrixSymbol('A', 3, 3)
  2238. assert _test_args(MatrixPermute(A, Permutation([2, 0, 1])))
  2239. def test_sympy__matrices__expressions__companion__CompanionMatrix():
  2240. from sympy.core.symbol import Symbol
  2241. from sympy.matrices.expressions.companion import CompanionMatrix
  2242. from sympy.polys.polytools import Poly
  2243. x = Symbol('x')
  2244. p = Poly([1, 2, 3], x)
  2245. assert _test_args(CompanionMatrix(p))
  2246. def test_sympy__physics__vector__frame__CoordinateSym():
  2247. from sympy.physics.vector import CoordinateSym
  2248. from sympy.physics.vector import ReferenceFrame
  2249. assert _test_args(CoordinateSym('R_x', ReferenceFrame('R'), 0))
  2250. def test_sympy__physics__paulialgebra__Pauli():
  2251. from sympy.physics.paulialgebra import Pauli
  2252. assert _test_args(Pauli(1))
  2253. def test_sympy__physics__quantum__anticommutator__AntiCommutator():
  2254. from sympy.physics.quantum.anticommutator import AntiCommutator
  2255. assert _test_args(AntiCommutator(x, y))
  2256. def test_sympy__physics__quantum__cartesian__PositionBra3D():
  2257. from sympy.physics.quantum.cartesian import PositionBra3D
  2258. assert _test_args(PositionBra3D(x, y, z))
  2259. def test_sympy__physics__quantum__cartesian__PositionKet3D():
  2260. from sympy.physics.quantum.cartesian import PositionKet3D
  2261. assert _test_args(PositionKet3D(x, y, z))
  2262. def test_sympy__physics__quantum__cartesian__PositionState3D():
  2263. from sympy.physics.quantum.cartesian import PositionState3D
  2264. assert _test_args(PositionState3D(x, y, z))
  2265. def test_sympy__physics__quantum__cartesian__PxBra():
  2266. from sympy.physics.quantum.cartesian import PxBra
  2267. assert _test_args(PxBra(x, y, z))
  2268. def test_sympy__physics__quantum__cartesian__PxKet():
  2269. from sympy.physics.quantum.cartesian import PxKet
  2270. assert _test_args(PxKet(x, y, z))
  2271. def test_sympy__physics__quantum__cartesian__PxOp():
  2272. from sympy.physics.quantum.cartesian import PxOp
  2273. assert _test_args(PxOp(x, y, z))
  2274. def test_sympy__physics__quantum__cartesian__XBra():
  2275. from sympy.physics.quantum.cartesian import XBra
  2276. assert _test_args(XBra(x))
  2277. def test_sympy__physics__quantum__cartesian__XKet():
  2278. from sympy.physics.quantum.cartesian import XKet
  2279. assert _test_args(XKet(x))
  2280. def test_sympy__physics__quantum__cartesian__XOp():
  2281. from sympy.physics.quantum.cartesian import XOp
  2282. assert _test_args(XOp(x))
  2283. def test_sympy__physics__quantum__cartesian__YOp():
  2284. from sympy.physics.quantum.cartesian import YOp
  2285. assert _test_args(YOp(x))
  2286. def test_sympy__physics__quantum__cartesian__ZOp():
  2287. from sympy.physics.quantum.cartesian import ZOp
  2288. assert _test_args(ZOp(x))
  2289. def test_sympy__physics__quantum__cg__CG():
  2290. from sympy.physics.quantum.cg import CG
  2291. from sympy.core.singleton import S
  2292. assert _test_args(CG(Rational(3, 2), Rational(3, 2), S.Half, Rational(-1, 2), 1, 1))
  2293. def test_sympy__physics__quantum__cg__Wigner3j():
  2294. from sympy.physics.quantum.cg import Wigner3j
  2295. assert _test_args(Wigner3j(6, 0, 4, 0, 2, 0))
  2296. def test_sympy__physics__quantum__cg__Wigner6j():
  2297. from sympy.physics.quantum.cg import Wigner6j
  2298. assert _test_args(Wigner6j(1, 2, 3, 2, 1, 2))
  2299. def test_sympy__physics__quantum__cg__Wigner9j():
  2300. from sympy.physics.quantum.cg import Wigner9j
  2301. assert _test_args(Wigner9j(2, 1, 1, Rational(3, 2), S.Half, 1, S.Half, S.Half, 0))
  2302. def test_sympy__physics__quantum__circuitplot__Mz():
  2303. from sympy.physics.quantum.circuitplot import Mz
  2304. assert _test_args(Mz(0))
  2305. def test_sympy__physics__quantum__circuitplot__Mx():
  2306. from sympy.physics.quantum.circuitplot import Mx
  2307. assert _test_args(Mx(0))
  2308. def test_sympy__physics__quantum__commutator__Commutator():
  2309. from sympy.physics.quantum.commutator import Commutator
  2310. A, B = symbols('A,B', commutative=False)
  2311. assert _test_args(Commutator(A, B))
  2312. def test_sympy__physics__quantum__constants__HBar():
  2313. from sympy.physics.quantum.constants import HBar
  2314. assert _test_args(HBar())
  2315. def test_sympy__physics__quantum__dagger__Dagger():
  2316. from sympy.physics.quantum.dagger import Dagger
  2317. from sympy.physics.quantum.state import Ket
  2318. assert _test_args(Dagger(Dagger(Ket('psi'))))
  2319. def test_sympy__physics__quantum__gate__CGate():
  2320. from sympy.physics.quantum.gate import CGate, Gate
  2321. assert _test_args(CGate((0, 1), Gate(2)))
  2322. def test_sympy__physics__quantum__gate__CGateS():
  2323. from sympy.physics.quantum.gate import CGateS, Gate
  2324. assert _test_args(CGateS((0, 1), Gate(2)))
  2325. def test_sympy__physics__quantum__gate__CNotGate():
  2326. from sympy.physics.quantum.gate import CNotGate
  2327. assert _test_args(CNotGate(0, 1))
  2328. def test_sympy__physics__quantum__gate__Gate():
  2329. from sympy.physics.quantum.gate import Gate
  2330. assert _test_args(Gate(0))
  2331. def test_sympy__physics__quantum__gate__HadamardGate():
  2332. from sympy.physics.quantum.gate import HadamardGate
  2333. assert _test_args(HadamardGate(0))
  2334. def test_sympy__physics__quantum__gate__IdentityGate():
  2335. from sympy.physics.quantum.gate import IdentityGate
  2336. assert _test_args(IdentityGate(0))
  2337. def test_sympy__physics__quantum__gate__OneQubitGate():
  2338. from sympy.physics.quantum.gate import OneQubitGate
  2339. assert _test_args(OneQubitGate(0))
  2340. def test_sympy__physics__quantum__gate__PhaseGate():
  2341. from sympy.physics.quantum.gate import PhaseGate
  2342. assert _test_args(PhaseGate(0))
  2343. def test_sympy__physics__quantum__gate__SwapGate():
  2344. from sympy.physics.quantum.gate import SwapGate
  2345. assert _test_args(SwapGate(0, 1))
  2346. def test_sympy__physics__quantum__gate__TGate():
  2347. from sympy.physics.quantum.gate import TGate
  2348. assert _test_args(TGate(0))
  2349. def test_sympy__physics__quantum__gate__TwoQubitGate():
  2350. from sympy.physics.quantum.gate import TwoQubitGate
  2351. assert _test_args(TwoQubitGate(0))
  2352. def test_sympy__physics__quantum__gate__UGate():
  2353. from sympy.physics.quantum.gate import UGate
  2354. from sympy.matrices.immutable import ImmutableDenseMatrix
  2355. from sympy.core.containers import Tuple
  2356. from sympy.core.numbers import Integer
  2357. assert _test_args(
  2358. UGate(Tuple(Integer(1)), ImmutableDenseMatrix([[1, 0], [0, 2]])))
  2359. def test_sympy__physics__quantum__gate__XGate():
  2360. from sympy.physics.quantum.gate import XGate
  2361. assert _test_args(XGate(0))
  2362. def test_sympy__physics__quantum__gate__YGate():
  2363. from sympy.physics.quantum.gate import YGate
  2364. assert _test_args(YGate(0))
  2365. def test_sympy__physics__quantum__gate__ZGate():
  2366. from sympy.physics.quantum.gate import ZGate
  2367. assert _test_args(ZGate(0))
  2368. def test_sympy__physics__quantum__grover__OracleGateFunction():
  2369. from sympy.physics.quantum.grover import OracleGateFunction
  2370. @OracleGateFunction
  2371. def f(qubit):
  2372. return
  2373. assert _test_args(f)
  2374. def test_sympy__physics__quantum__grover__OracleGate():
  2375. from sympy.physics.quantum.grover import OracleGate
  2376. def f(qubit):
  2377. return
  2378. assert _test_args(OracleGate(1,f))
  2379. def test_sympy__physics__quantum__grover__WGate():
  2380. from sympy.physics.quantum.grover import WGate
  2381. assert _test_args(WGate(1))
  2382. def test_sympy__physics__quantum__hilbert__ComplexSpace():
  2383. from sympy.physics.quantum.hilbert import ComplexSpace
  2384. assert _test_args(ComplexSpace(x))
  2385. def test_sympy__physics__quantum__hilbert__DirectSumHilbertSpace():
  2386. from sympy.physics.quantum.hilbert import DirectSumHilbertSpace, ComplexSpace, FockSpace
  2387. c = ComplexSpace(2)
  2388. f = FockSpace()
  2389. assert _test_args(DirectSumHilbertSpace(c, f))
  2390. def test_sympy__physics__quantum__hilbert__FockSpace():
  2391. from sympy.physics.quantum.hilbert import FockSpace
  2392. assert _test_args(FockSpace())
  2393. def test_sympy__physics__quantum__hilbert__HilbertSpace():
  2394. from sympy.physics.quantum.hilbert import HilbertSpace
  2395. assert _test_args(HilbertSpace())
  2396. def test_sympy__physics__quantum__hilbert__L2():
  2397. from sympy.physics.quantum.hilbert import L2
  2398. from sympy.core.numbers import oo
  2399. from sympy.sets.sets import Interval
  2400. assert _test_args(L2(Interval(0, oo)))
  2401. def test_sympy__physics__quantum__hilbert__TensorPowerHilbertSpace():
  2402. from sympy.physics.quantum.hilbert import TensorPowerHilbertSpace, FockSpace
  2403. f = FockSpace()
  2404. assert _test_args(TensorPowerHilbertSpace(f, 2))
  2405. def test_sympy__physics__quantum__hilbert__TensorProductHilbertSpace():
  2406. from sympy.physics.quantum.hilbert import TensorProductHilbertSpace, FockSpace, ComplexSpace
  2407. c = ComplexSpace(2)
  2408. f = FockSpace()
  2409. assert _test_args(TensorProductHilbertSpace(f, c))
  2410. def test_sympy__physics__quantum__innerproduct__InnerProduct():
  2411. from sympy.physics.quantum import Bra, Ket, InnerProduct
  2412. b = Bra('b')
  2413. k = Ket('k')
  2414. assert _test_args(InnerProduct(b, k))
  2415. def test_sympy__physics__quantum__operator__DifferentialOperator():
  2416. from sympy.physics.quantum.operator import DifferentialOperator
  2417. from sympy.core.function import (Derivative, Function)
  2418. f = Function('f')
  2419. assert _test_args(DifferentialOperator(1/x*Derivative(f(x), x), f(x)))
  2420. def test_sympy__physics__quantum__operator__HermitianOperator():
  2421. from sympy.physics.quantum.operator import HermitianOperator
  2422. assert _test_args(HermitianOperator('H'))
  2423. def test_sympy__physics__quantum__operator__IdentityOperator():
  2424. from sympy.physics.quantum.operator import IdentityOperator
  2425. assert _test_args(IdentityOperator(5))
  2426. def test_sympy__physics__quantum__operator__Operator():
  2427. from sympy.physics.quantum.operator import Operator
  2428. assert _test_args(Operator('A'))
  2429. def test_sympy__physics__quantum__operator__OuterProduct():
  2430. from sympy.physics.quantum.operator import OuterProduct
  2431. from sympy.physics.quantum import Ket, Bra
  2432. b = Bra('b')
  2433. k = Ket('k')
  2434. assert _test_args(OuterProduct(k, b))
  2435. def test_sympy__physics__quantum__operator__UnitaryOperator():
  2436. from sympy.physics.quantum.operator import UnitaryOperator
  2437. assert _test_args(UnitaryOperator('U'))
  2438. def test_sympy__physics__quantum__piab__PIABBra():
  2439. from sympy.physics.quantum.piab import PIABBra
  2440. assert _test_args(PIABBra('B'))
  2441. def test_sympy__physics__quantum__boson__BosonOp():
  2442. from sympy.physics.quantum.boson import BosonOp
  2443. assert _test_args(BosonOp('a'))
  2444. assert _test_args(BosonOp('a', False))
  2445. def test_sympy__physics__quantum__boson__BosonFockKet():
  2446. from sympy.physics.quantum.boson import BosonFockKet
  2447. assert _test_args(BosonFockKet(1))
  2448. def test_sympy__physics__quantum__boson__BosonFockBra():
  2449. from sympy.physics.quantum.boson import BosonFockBra
  2450. assert _test_args(BosonFockBra(1))
  2451. def test_sympy__physics__quantum__boson__BosonCoherentKet():
  2452. from sympy.physics.quantum.boson import BosonCoherentKet
  2453. assert _test_args(BosonCoherentKet(1))
  2454. def test_sympy__physics__quantum__boson__BosonCoherentBra():
  2455. from sympy.physics.quantum.boson import BosonCoherentBra
  2456. assert _test_args(BosonCoherentBra(1))
  2457. def test_sympy__physics__quantum__fermion__FermionOp():
  2458. from sympy.physics.quantum.fermion import FermionOp
  2459. assert _test_args(FermionOp('c'))
  2460. assert _test_args(FermionOp('c', False))
  2461. def test_sympy__physics__quantum__fermion__FermionFockKet():
  2462. from sympy.physics.quantum.fermion import FermionFockKet
  2463. assert _test_args(FermionFockKet(1))
  2464. def test_sympy__physics__quantum__fermion__FermionFockBra():
  2465. from sympy.physics.quantum.fermion import FermionFockBra
  2466. assert _test_args(FermionFockBra(1))
  2467. def test_sympy__physics__quantum__pauli__SigmaOpBase():
  2468. from sympy.physics.quantum.pauli import SigmaOpBase
  2469. assert _test_args(SigmaOpBase())
  2470. def test_sympy__physics__quantum__pauli__SigmaX():
  2471. from sympy.physics.quantum.pauli import SigmaX
  2472. assert _test_args(SigmaX())
  2473. def test_sympy__physics__quantum__pauli__SigmaY():
  2474. from sympy.physics.quantum.pauli import SigmaY
  2475. assert _test_args(SigmaY())
  2476. def test_sympy__physics__quantum__pauli__SigmaZ():
  2477. from sympy.physics.quantum.pauli import SigmaZ
  2478. assert _test_args(SigmaZ())
  2479. def test_sympy__physics__quantum__pauli__SigmaMinus():
  2480. from sympy.physics.quantum.pauli import SigmaMinus
  2481. assert _test_args(SigmaMinus())
  2482. def test_sympy__physics__quantum__pauli__SigmaPlus():
  2483. from sympy.physics.quantum.pauli import SigmaPlus
  2484. assert _test_args(SigmaPlus())
  2485. def test_sympy__physics__quantum__pauli__SigmaZKet():
  2486. from sympy.physics.quantum.pauli import SigmaZKet
  2487. assert _test_args(SigmaZKet(0))
  2488. def test_sympy__physics__quantum__pauli__SigmaZBra():
  2489. from sympy.physics.quantum.pauli import SigmaZBra
  2490. assert _test_args(SigmaZBra(0))
  2491. def test_sympy__physics__quantum__piab__PIABHamiltonian():
  2492. from sympy.physics.quantum.piab import PIABHamiltonian
  2493. assert _test_args(PIABHamiltonian('P'))
  2494. def test_sympy__physics__quantum__piab__PIABKet():
  2495. from sympy.physics.quantum.piab import PIABKet
  2496. assert _test_args(PIABKet('K'))
  2497. def test_sympy__physics__quantum__qexpr__QExpr():
  2498. from sympy.physics.quantum.qexpr import QExpr
  2499. assert _test_args(QExpr(0))
  2500. def test_sympy__physics__quantum__qft__Fourier():
  2501. from sympy.physics.quantum.qft import Fourier
  2502. assert _test_args(Fourier(0, 1))
  2503. def test_sympy__physics__quantum__qft__IQFT():
  2504. from sympy.physics.quantum.qft import IQFT
  2505. assert _test_args(IQFT(0, 1))
  2506. def test_sympy__physics__quantum__qft__QFT():
  2507. from sympy.physics.quantum.qft import QFT
  2508. assert _test_args(QFT(0, 1))
  2509. def test_sympy__physics__quantum__qft__RkGate():
  2510. from sympy.physics.quantum.qft import RkGate
  2511. assert _test_args(RkGate(0, 1))
  2512. def test_sympy__physics__quantum__qubit__IntQubit():
  2513. from sympy.physics.quantum.qubit import IntQubit
  2514. assert _test_args(IntQubit(0))
  2515. def test_sympy__physics__quantum__qubit__IntQubitBra():
  2516. from sympy.physics.quantum.qubit import IntQubitBra
  2517. assert _test_args(IntQubitBra(0))
  2518. def test_sympy__physics__quantum__qubit__IntQubitState():
  2519. from sympy.physics.quantum.qubit import IntQubitState, QubitState
  2520. assert _test_args(IntQubitState(QubitState(0, 1)))
  2521. def test_sympy__physics__quantum__qubit__Qubit():
  2522. from sympy.physics.quantum.qubit import Qubit
  2523. assert _test_args(Qubit(0, 0, 0))
  2524. def test_sympy__physics__quantum__qubit__QubitBra():
  2525. from sympy.physics.quantum.qubit import QubitBra
  2526. assert _test_args(QubitBra('1', 0))
  2527. def test_sympy__physics__quantum__qubit__QubitState():
  2528. from sympy.physics.quantum.qubit import QubitState
  2529. assert _test_args(QubitState(0, 1))
  2530. def test_sympy__physics__quantum__density__Density():
  2531. from sympy.physics.quantum.density import Density
  2532. from sympy.physics.quantum.state import Ket
  2533. assert _test_args(Density([Ket(0), 0.5], [Ket(1), 0.5]))
  2534. @SKIP("TODO: sympy.physics.quantum.shor: Cmod Not Implemented")
  2535. def test_sympy__physics__quantum__shor__CMod():
  2536. from sympy.physics.quantum.shor import CMod
  2537. assert _test_args(CMod())
  2538. def test_sympy__physics__quantum__spin__CoupledSpinState():
  2539. from sympy.physics.quantum.spin import CoupledSpinState
  2540. assert _test_args(CoupledSpinState(1, 0, (1, 1)))
  2541. assert _test_args(CoupledSpinState(1, 0, (1, S.Half, S.Half)))
  2542. assert _test_args(CoupledSpinState(
  2543. 1, 0, (1, S.Half, S.Half), ((2, 3, S.Half), (1, 2, 1)) ))
  2544. j, m, j1, j2, j3, j12, x = symbols('j m j1:4 j12 x')
  2545. assert CoupledSpinState(
  2546. j, m, (j1, j2, j3)).subs(j2, x) == CoupledSpinState(j, m, (j1, x, j3))
  2547. assert CoupledSpinState(j, m, (j1, j2, j3), ((1, 3, j12), (1, 2, j)) ).subs(j12, x) == \
  2548. CoupledSpinState(j, m, (j1, j2, j3), ((1, 3, x), (1, 2, j)) )
  2549. def test_sympy__physics__quantum__spin__J2Op():
  2550. from sympy.physics.quantum.spin import J2Op
  2551. assert _test_args(J2Op('J'))
  2552. def test_sympy__physics__quantum__spin__JminusOp():
  2553. from sympy.physics.quantum.spin import JminusOp
  2554. assert _test_args(JminusOp('J'))
  2555. def test_sympy__physics__quantum__spin__JplusOp():
  2556. from sympy.physics.quantum.spin import JplusOp
  2557. assert _test_args(JplusOp('J'))
  2558. def test_sympy__physics__quantum__spin__JxBra():
  2559. from sympy.physics.quantum.spin import JxBra
  2560. assert _test_args(JxBra(1, 0))
  2561. def test_sympy__physics__quantum__spin__JxBraCoupled():
  2562. from sympy.physics.quantum.spin import JxBraCoupled
  2563. assert _test_args(JxBraCoupled(1, 0, (1, 1)))
  2564. def test_sympy__physics__quantum__spin__JxKet():
  2565. from sympy.physics.quantum.spin import JxKet
  2566. assert _test_args(JxKet(1, 0))
  2567. def test_sympy__physics__quantum__spin__JxKetCoupled():
  2568. from sympy.physics.quantum.spin import JxKetCoupled
  2569. assert _test_args(JxKetCoupled(1, 0, (1, 1)))
  2570. def test_sympy__physics__quantum__spin__JxOp():
  2571. from sympy.physics.quantum.spin import JxOp
  2572. assert _test_args(JxOp('J'))
  2573. def test_sympy__physics__quantum__spin__JyBra():
  2574. from sympy.physics.quantum.spin import JyBra
  2575. assert _test_args(JyBra(1, 0))
  2576. def test_sympy__physics__quantum__spin__JyBraCoupled():
  2577. from sympy.physics.quantum.spin import JyBraCoupled
  2578. assert _test_args(JyBraCoupled(1, 0, (1, 1)))
  2579. def test_sympy__physics__quantum__spin__JyKet():
  2580. from sympy.physics.quantum.spin import JyKet
  2581. assert _test_args(JyKet(1, 0))
  2582. def test_sympy__physics__quantum__spin__JyKetCoupled():
  2583. from sympy.physics.quantum.spin import JyKetCoupled
  2584. assert _test_args(JyKetCoupled(1, 0, (1, 1)))
  2585. def test_sympy__physics__quantum__spin__JyOp():
  2586. from sympy.physics.quantum.spin import JyOp
  2587. assert _test_args(JyOp('J'))
  2588. def test_sympy__physics__quantum__spin__JzBra():
  2589. from sympy.physics.quantum.spin import JzBra
  2590. assert _test_args(JzBra(1, 0))
  2591. def test_sympy__physics__quantum__spin__JzBraCoupled():
  2592. from sympy.physics.quantum.spin import JzBraCoupled
  2593. assert _test_args(JzBraCoupled(1, 0, (1, 1)))
  2594. def test_sympy__physics__quantum__spin__JzKet():
  2595. from sympy.physics.quantum.spin import JzKet
  2596. assert _test_args(JzKet(1, 0))
  2597. def test_sympy__physics__quantum__spin__JzKetCoupled():
  2598. from sympy.physics.quantum.spin import JzKetCoupled
  2599. assert _test_args(JzKetCoupled(1, 0, (1, 1)))
  2600. def test_sympy__physics__quantum__spin__JzOp():
  2601. from sympy.physics.quantum.spin import JzOp
  2602. assert _test_args(JzOp('J'))
  2603. def test_sympy__physics__quantum__spin__Rotation():
  2604. from sympy.physics.quantum.spin import Rotation
  2605. assert _test_args(Rotation(pi, 0, pi/2))
  2606. def test_sympy__physics__quantum__spin__SpinState():
  2607. from sympy.physics.quantum.spin import SpinState
  2608. assert _test_args(SpinState(1, 0))
  2609. def test_sympy__physics__quantum__spin__WignerD():
  2610. from sympy.physics.quantum.spin import WignerD
  2611. assert _test_args(WignerD(0, 1, 2, 3, 4, 5))
  2612. def test_sympy__physics__quantum__state__Bra():
  2613. from sympy.physics.quantum.state import Bra
  2614. assert _test_args(Bra(0))
  2615. def test_sympy__physics__quantum__state__BraBase():
  2616. from sympy.physics.quantum.state import BraBase
  2617. assert _test_args(BraBase(0))
  2618. def test_sympy__physics__quantum__state__Ket():
  2619. from sympy.physics.quantum.state import Ket
  2620. assert _test_args(Ket(0))
  2621. def test_sympy__physics__quantum__state__KetBase():
  2622. from sympy.physics.quantum.state import KetBase
  2623. assert _test_args(KetBase(0))
  2624. def test_sympy__physics__quantum__state__State():
  2625. from sympy.physics.quantum.state import State
  2626. assert _test_args(State(0))
  2627. def test_sympy__physics__quantum__state__StateBase():
  2628. from sympy.physics.quantum.state import StateBase
  2629. assert _test_args(StateBase(0))
  2630. def test_sympy__physics__quantum__state__OrthogonalBra():
  2631. from sympy.physics.quantum.state import OrthogonalBra
  2632. assert _test_args(OrthogonalBra(0))
  2633. def test_sympy__physics__quantum__state__OrthogonalKet():
  2634. from sympy.physics.quantum.state import OrthogonalKet
  2635. assert _test_args(OrthogonalKet(0))
  2636. def test_sympy__physics__quantum__state__OrthogonalState():
  2637. from sympy.physics.quantum.state import OrthogonalState
  2638. assert _test_args(OrthogonalState(0))
  2639. def test_sympy__physics__quantum__state__TimeDepBra():
  2640. from sympy.physics.quantum.state import TimeDepBra
  2641. assert _test_args(TimeDepBra('psi', 't'))
  2642. def test_sympy__physics__quantum__state__TimeDepKet():
  2643. from sympy.physics.quantum.state import TimeDepKet
  2644. assert _test_args(TimeDepKet('psi', 't'))
  2645. def test_sympy__physics__quantum__state__TimeDepState():
  2646. from sympy.physics.quantum.state import TimeDepState
  2647. assert _test_args(TimeDepState('psi', 't'))
  2648. def test_sympy__physics__quantum__state__Wavefunction():
  2649. from sympy.physics.quantum.state import Wavefunction
  2650. from sympy.functions import sin
  2651. from sympy.functions.elementary.piecewise import Piecewise
  2652. n = 1
  2653. L = 1
  2654. g = Piecewise((0, x < 0), (0, x > L), (sqrt(2//L)*sin(n*pi*x/L), True))
  2655. assert _test_args(Wavefunction(g, x))
  2656. def test_sympy__physics__quantum__tensorproduct__TensorProduct():
  2657. from sympy.physics.quantum.tensorproduct import TensorProduct
  2658. x, y = symbols("x y", commutative=False)
  2659. assert _test_args(TensorProduct(x, y))
  2660. def test_sympy__physics__quantum__identitysearch__GateIdentity():
  2661. from sympy.physics.quantum.gate import X
  2662. from sympy.physics.quantum.identitysearch import GateIdentity
  2663. assert _test_args(GateIdentity(X(0), X(0)))
  2664. def test_sympy__physics__quantum__sho1d__SHOOp():
  2665. from sympy.physics.quantum.sho1d import SHOOp
  2666. assert _test_args(SHOOp('a'))
  2667. def test_sympy__physics__quantum__sho1d__RaisingOp():
  2668. from sympy.physics.quantum.sho1d import RaisingOp
  2669. assert _test_args(RaisingOp('a'))
  2670. def test_sympy__physics__quantum__sho1d__LoweringOp():
  2671. from sympy.physics.quantum.sho1d import LoweringOp
  2672. assert _test_args(LoweringOp('a'))
  2673. def test_sympy__physics__quantum__sho1d__NumberOp():
  2674. from sympy.physics.quantum.sho1d import NumberOp
  2675. assert _test_args(NumberOp('N'))
  2676. def test_sympy__physics__quantum__sho1d__Hamiltonian():
  2677. from sympy.physics.quantum.sho1d import Hamiltonian
  2678. assert _test_args(Hamiltonian('H'))
  2679. def test_sympy__physics__quantum__sho1d__SHOState():
  2680. from sympy.physics.quantum.sho1d import SHOState
  2681. assert _test_args(SHOState(0))
  2682. def test_sympy__physics__quantum__sho1d__SHOKet():
  2683. from sympy.physics.quantum.sho1d import SHOKet
  2684. assert _test_args(SHOKet(0))
  2685. def test_sympy__physics__quantum__sho1d__SHOBra():
  2686. from sympy.physics.quantum.sho1d import SHOBra
  2687. assert _test_args(SHOBra(0))
  2688. def test_sympy__physics__secondquant__AnnihilateBoson():
  2689. from sympy.physics.secondquant import AnnihilateBoson
  2690. assert _test_args(AnnihilateBoson(0))
  2691. def test_sympy__physics__secondquant__AnnihilateFermion():
  2692. from sympy.physics.secondquant import AnnihilateFermion
  2693. assert _test_args(AnnihilateFermion(0))
  2694. @SKIP("abstract class")
  2695. def test_sympy__physics__secondquant__Annihilator():
  2696. pass
  2697. def test_sympy__physics__secondquant__AntiSymmetricTensor():
  2698. from sympy.physics.secondquant import AntiSymmetricTensor
  2699. i, j = symbols('i j', below_fermi=True)
  2700. a, b = symbols('a b', above_fermi=True)
  2701. assert _test_args(AntiSymmetricTensor('v', (a, i), (b, j)))
  2702. def test_sympy__physics__secondquant__BosonState():
  2703. from sympy.physics.secondquant import BosonState
  2704. assert _test_args(BosonState((0, 1)))
  2705. @SKIP("abstract class")
  2706. def test_sympy__physics__secondquant__BosonicOperator():
  2707. pass
  2708. def test_sympy__physics__secondquant__Commutator():
  2709. from sympy.physics.secondquant import Commutator
  2710. x, y = symbols('x y', commutative=False)
  2711. assert _test_args(Commutator(x, y))
  2712. def test_sympy__physics__secondquant__CreateBoson():
  2713. from sympy.physics.secondquant import CreateBoson
  2714. assert _test_args(CreateBoson(0))
  2715. def test_sympy__physics__secondquant__CreateFermion():
  2716. from sympy.physics.secondquant import CreateFermion
  2717. assert _test_args(CreateFermion(0))
  2718. @SKIP("abstract class")
  2719. def test_sympy__physics__secondquant__Creator():
  2720. pass
  2721. def test_sympy__physics__secondquant__Dagger():
  2722. from sympy.physics.secondquant import Dagger
  2723. assert _test_args(Dagger(x))
  2724. def test_sympy__physics__secondquant__FermionState():
  2725. from sympy.physics.secondquant import FermionState
  2726. assert _test_args(FermionState((0, 1)))
  2727. def test_sympy__physics__secondquant__FermionicOperator():
  2728. from sympy.physics.secondquant import FermionicOperator
  2729. assert _test_args(FermionicOperator(0))
  2730. def test_sympy__physics__secondquant__FockState():
  2731. from sympy.physics.secondquant import FockState
  2732. assert _test_args(FockState((0, 1)))
  2733. def test_sympy__physics__secondquant__FockStateBosonBra():
  2734. from sympy.physics.secondquant import FockStateBosonBra
  2735. assert _test_args(FockStateBosonBra((0, 1)))
  2736. def test_sympy__physics__secondquant__FockStateBosonKet():
  2737. from sympy.physics.secondquant import FockStateBosonKet
  2738. assert _test_args(FockStateBosonKet((0, 1)))
  2739. def test_sympy__physics__secondquant__FockStateBra():
  2740. from sympy.physics.secondquant import FockStateBra
  2741. assert _test_args(FockStateBra((0, 1)))
  2742. def test_sympy__physics__secondquant__FockStateFermionBra():
  2743. from sympy.physics.secondquant import FockStateFermionBra
  2744. assert _test_args(FockStateFermionBra((0, 1)))
  2745. def test_sympy__physics__secondquant__FockStateFermionKet():
  2746. from sympy.physics.secondquant import FockStateFermionKet
  2747. assert _test_args(FockStateFermionKet((0, 1)))
  2748. def test_sympy__physics__secondquant__FockStateKet():
  2749. from sympy.physics.secondquant import FockStateKet
  2750. assert _test_args(FockStateKet((0, 1)))
  2751. def test_sympy__physics__secondquant__InnerProduct():
  2752. from sympy.physics.secondquant import InnerProduct
  2753. from sympy.physics.secondquant import FockStateKet, FockStateBra
  2754. assert _test_args(InnerProduct(FockStateBra((0, 1)), FockStateKet((0, 1))))
  2755. def test_sympy__physics__secondquant__NO():
  2756. from sympy.physics.secondquant import NO, F, Fd
  2757. assert _test_args(NO(Fd(x)*F(y)))
  2758. def test_sympy__physics__secondquant__PermutationOperator():
  2759. from sympy.physics.secondquant import PermutationOperator
  2760. assert _test_args(PermutationOperator(0, 1))
  2761. def test_sympy__physics__secondquant__SqOperator():
  2762. from sympy.physics.secondquant import SqOperator
  2763. assert _test_args(SqOperator(0))
  2764. def test_sympy__physics__secondquant__TensorSymbol():
  2765. from sympy.physics.secondquant import TensorSymbol
  2766. assert _test_args(TensorSymbol(x))
  2767. def test_sympy__physics__control__lti__LinearTimeInvariant():
  2768. # Direct instances of LinearTimeInvariant class are not allowed.
  2769. # func(*args) tests for its derived classes (TransferFunction,
  2770. # Series, Parallel and TransferFunctionMatrix) should pass.
  2771. pass
  2772. def test_sympy__physics__control__lti__SISOLinearTimeInvariant():
  2773. # Direct instances of SISOLinearTimeInvariant class are not allowed.
  2774. pass
  2775. def test_sympy__physics__control__lti__MIMOLinearTimeInvariant():
  2776. # Direct instances of MIMOLinearTimeInvariant class are not allowed.
  2777. pass
  2778. def test_sympy__physics__control__lti__TransferFunction():
  2779. from sympy.physics.control.lti import TransferFunction
  2780. assert _test_args(TransferFunction(2, 3, x))
  2781. def test_sympy__physics__control__lti__Series():
  2782. from sympy.physics.control import Series, TransferFunction
  2783. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2784. tf2 = TransferFunction(y - x, z + y, x)
  2785. assert _test_args(Series(tf1, tf2))
  2786. def test_sympy__physics__control__lti__MIMOSeries():
  2787. from sympy.physics.control import MIMOSeries, TransferFunction, TransferFunctionMatrix
  2788. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2789. tf2 = TransferFunction(y - x, z + y, x)
  2790. tfm_1 = TransferFunctionMatrix([[tf2, tf1]])
  2791. tfm_2 = TransferFunctionMatrix([[tf1, tf2], [tf2, tf1]])
  2792. tfm_3 = TransferFunctionMatrix([[tf1], [tf2]])
  2793. assert _test_args(MIMOSeries(tfm_3, tfm_2, tfm_1))
  2794. def test_sympy__physics__control__lti__Parallel():
  2795. from sympy.physics.control import Parallel, TransferFunction
  2796. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2797. tf2 = TransferFunction(y - x, z + y, x)
  2798. assert _test_args(Parallel(tf1, tf2))
  2799. def test_sympy__physics__control__lti__MIMOParallel():
  2800. from sympy.physics.control import MIMOParallel, TransferFunction, TransferFunctionMatrix
  2801. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2802. tf2 = TransferFunction(y - x, z + y, x)
  2803. tfm_1 = TransferFunctionMatrix([[tf1, tf2], [tf2, tf1]])
  2804. tfm_2 = TransferFunctionMatrix([[tf2, tf1], [tf1, tf2]])
  2805. assert _test_args(MIMOParallel(tfm_1, tfm_2))
  2806. def test_sympy__physics__control__lti__Feedback():
  2807. from sympy.physics.control import TransferFunction, Feedback
  2808. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2809. tf2 = TransferFunction(y - x, z + y, x)
  2810. assert _test_args(Feedback(tf1, tf2))
  2811. assert _test_args(Feedback(tf1, tf2, 1))
  2812. def test_sympy__physics__control__lti__MIMOFeedback():
  2813. from sympy.physics.control import TransferFunction, MIMOFeedback, TransferFunctionMatrix
  2814. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2815. tf2 = TransferFunction(y - x, z + y, x)
  2816. tfm_1 = TransferFunctionMatrix([[tf2, tf1], [tf1, tf2]])
  2817. tfm_2 = TransferFunctionMatrix([[tf1, tf2], [tf2, tf1]])
  2818. assert _test_args(MIMOFeedback(tfm_1, tfm_2))
  2819. assert _test_args(MIMOFeedback(tfm_1, tfm_2, 1))
  2820. def test_sympy__physics__control__lti__TransferFunctionMatrix():
  2821. from sympy.physics.control import TransferFunction, TransferFunctionMatrix
  2822. tf1 = TransferFunction(x**2 - y**3, y - z, x)
  2823. tf2 = TransferFunction(y - x, z + y, x)
  2824. assert _test_args(TransferFunctionMatrix([[tf1, tf2]]))
  2825. def test_sympy__physics__units__dimensions__Dimension():
  2826. from sympy.physics.units.dimensions import Dimension
  2827. assert _test_args(Dimension("length", "L"))
  2828. def test_sympy__physics__units__dimensions__DimensionSystem():
  2829. from sympy.physics.units.dimensions import DimensionSystem
  2830. from sympy.physics.units.definitions.dimension_definitions import length, time, velocity
  2831. assert _test_args(DimensionSystem((length, time), (velocity,)))
  2832. def test_sympy__physics__units__quantities__Quantity():
  2833. from sympy.physics.units.quantities import Quantity
  2834. assert _test_args(Quantity("dam"))
  2835. def test_sympy__physics__units__quantities__PhysicalConstant():
  2836. from sympy.physics.units.quantities import PhysicalConstant
  2837. assert _test_args(PhysicalConstant("foo"))
  2838. def test_sympy__physics__units__prefixes__Prefix():
  2839. from sympy.physics.units.prefixes import Prefix
  2840. assert _test_args(Prefix('kilo', 'k', 3))
  2841. def test_sympy__core__numbers__AlgebraicNumber():
  2842. from sympy.core.numbers import AlgebraicNumber
  2843. assert _test_args(AlgebraicNumber(sqrt(2), [1, 2, 3]))
  2844. def test_sympy__polys__polytools__GroebnerBasis():
  2845. from sympy.polys.polytools import GroebnerBasis
  2846. assert _test_args(GroebnerBasis([x, y, z], x, y, z))
  2847. def test_sympy__polys__polytools__Poly():
  2848. from sympy.polys.polytools import Poly
  2849. assert _test_args(Poly(2, x, y))
  2850. def test_sympy__polys__polytools__PurePoly():
  2851. from sympy.polys.polytools import PurePoly
  2852. assert _test_args(PurePoly(2, x, y))
  2853. @SKIP('abstract class')
  2854. def test_sympy__polys__rootoftools__RootOf():
  2855. pass
  2856. def test_sympy__polys__rootoftools__ComplexRootOf():
  2857. from sympy.polys.rootoftools import ComplexRootOf
  2858. assert _test_args(ComplexRootOf(x**3 + x + 1, 0))
  2859. def test_sympy__polys__rootoftools__RootSum():
  2860. from sympy.polys.rootoftools import RootSum
  2861. assert _test_args(RootSum(x**3 + x + 1, sin))
  2862. def test_sympy__series__limits__Limit():
  2863. from sympy.series.limits import Limit
  2864. assert _test_args(Limit(x, x, 0, dir='-'))
  2865. def test_sympy__series__order__Order():
  2866. from sympy.series.order import Order
  2867. assert _test_args(Order(1, x, y))
  2868. @SKIP('Abstract Class')
  2869. def test_sympy__series__sequences__SeqBase():
  2870. pass
  2871. def test_sympy__series__sequences__EmptySequence():
  2872. # Need to import the instance from series not the class from
  2873. # series.sequence
  2874. from sympy.series import EmptySequence
  2875. assert _test_args(EmptySequence)
  2876. @SKIP('Abstract Class')
  2877. def test_sympy__series__sequences__SeqExpr():
  2878. pass
  2879. def test_sympy__series__sequences__SeqPer():
  2880. from sympy.series.sequences import SeqPer
  2881. assert _test_args(SeqPer((1, 2, 3), (0, 10)))
  2882. def test_sympy__series__sequences__SeqFormula():
  2883. from sympy.series.sequences import SeqFormula
  2884. assert _test_args(SeqFormula(x**2, (0, 10)))
  2885. def test_sympy__series__sequences__RecursiveSeq():
  2886. from sympy.series.sequences import RecursiveSeq
  2887. y = Function("y")
  2888. n = symbols("n")
  2889. assert _test_args(RecursiveSeq(y(n - 1) + y(n - 2), y(n), n, (0, 1)))
  2890. assert _test_args(RecursiveSeq(y(n - 1) + y(n - 2), y(n), n))
  2891. def test_sympy__series__sequences__SeqExprOp():
  2892. from sympy.series.sequences import SeqExprOp, sequence
  2893. s1 = sequence((1, 2, 3))
  2894. s2 = sequence(x**2)
  2895. assert _test_args(SeqExprOp(s1, s2))
  2896. def test_sympy__series__sequences__SeqAdd():
  2897. from sympy.series.sequences import SeqAdd, sequence
  2898. s1 = sequence((1, 2, 3))
  2899. s2 = sequence(x**2)
  2900. assert _test_args(SeqAdd(s1, s2))
  2901. def test_sympy__series__sequences__SeqMul():
  2902. from sympy.series.sequences import SeqMul, sequence
  2903. s1 = sequence((1, 2, 3))
  2904. s2 = sequence(x**2)
  2905. assert _test_args(SeqMul(s1, s2))
  2906. @SKIP('Abstract Class')
  2907. def test_sympy__series__series_class__SeriesBase():
  2908. pass
  2909. def test_sympy__series__fourier__FourierSeries():
  2910. from sympy.series.fourier import fourier_series
  2911. assert _test_args(fourier_series(x, (x, -pi, pi)))
  2912. def test_sympy__series__fourier__FiniteFourierSeries():
  2913. from sympy.series.fourier import fourier_series
  2914. assert _test_args(fourier_series(sin(pi*x), (x, -1, 1)))
  2915. def test_sympy__series__formal__FormalPowerSeries():
  2916. from sympy.series.formal import fps
  2917. assert _test_args(fps(log(1 + x), x))
  2918. def test_sympy__series__formal__Coeff():
  2919. from sympy.series.formal import fps
  2920. assert _test_args(fps(x**2 + x + 1, x))
  2921. @SKIP('Abstract Class')
  2922. def test_sympy__series__formal__FiniteFormalPowerSeries():
  2923. pass
  2924. def test_sympy__series__formal__FormalPowerSeriesProduct():
  2925. from sympy.series.formal import fps
  2926. f1, f2 = fps(sin(x)), fps(exp(x))
  2927. assert _test_args(f1.product(f2, x))
  2928. def test_sympy__series__formal__FormalPowerSeriesCompose():
  2929. from sympy.series.formal import fps
  2930. f1, f2 = fps(exp(x)), fps(sin(x))
  2931. assert _test_args(f1.compose(f2, x))
  2932. def test_sympy__series__formal__FormalPowerSeriesInverse():
  2933. from sympy.series.formal import fps
  2934. f1 = fps(exp(x))
  2935. assert _test_args(f1.inverse(x))
  2936. def test_sympy__simplify__hyperexpand__Hyper_Function():
  2937. from sympy.simplify.hyperexpand import Hyper_Function
  2938. assert _test_args(Hyper_Function([2], [1]))
  2939. def test_sympy__simplify__hyperexpand__G_Function():
  2940. from sympy.simplify.hyperexpand import G_Function
  2941. assert _test_args(G_Function([2], [1], [], []))
  2942. @SKIP("abstract class")
  2943. def test_sympy__tensor__array__ndim_array__ImmutableNDimArray():
  2944. pass
  2945. def test_sympy__tensor__array__dense_ndim_array__ImmutableDenseNDimArray():
  2946. from sympy.tensor.array.dense_ndim_array import ImmutableDenseNDimArray
  2947. densarr = ImmutableDenseNDimArray(range(10, 34), (2, 3, 4))
  2948. assert _test_args(densarr)
  2949. def test_sympy__tensor__array__sparse_ndim_array__ImmutableSparseNDimArray():
  2950. from sympy.tensor.array.sparse_ndim_array import ImmutableSparseNDimArray
  2951. sparr = ImmutableSparseNDimArray(range(10, 34), (2, 3, 4))
  2952. assert _test_args(sparr)
  2953. def test_sympy__tensor__array__array_comprehension__ArrayComprehension():
  2954. from sympy.tensor.array.array_comprehension import ArrayComprehension
  2955. arrcom = ArrayComprehension(x, (x, 1, 5))
  2956. assert _test_args(arrcom)
  2957. def test_sympy__tensor__array__array_comprehension__ArrayComprehensionMap():
  2958. from sympy.tensor.array.array_comprehension import ArrayComprehensionMap
  2959. arrcomma = ArrayComprehensionMap(lambda: 0, (x, 1, 5))
  2960. assert _test_args(arrcomma)
  2961. def test_sympy__tensor__array__array_derivatives__ArrayDerivative():
  2962. from sympy.tensor.array.array_derivatives import ArrayDerivative
  2963. A = MatrixSymbol("A", 2, 2)
  2964. arrder = ArrayDerivative(A, A, evaluate=False)
  2965. assert _test_args(arrder)
  2966. def test_sympy__tensor__array__expressions__array_expressions__ArraySymbol():
  2967. from sympy.tensor.array.expressions.array_expressions import ArraySymbol
  2968. m, n, k = symbols("m n k")
  2969. array = ArraySymbol("A", (m, n, k, 2))
  2970. assert _test_args(array)
  2971. def test_sympy__tensor__array__expressions__array_expressions__ArrayElement():
  2972. from sympy.tensor.array.expressions.array_expressions import ArrayElement
  2973. m, n, k = symbols("m n k")
  2974. ae = ArrayElement("A", (m, n, k, 2))
  2975. assert _test_args(ae)
  2976. def test_sympy__tensor__array__expressions__array_expressions__ZeroArray():
  2977. from sympy.tensor.array.expressions.array_expressions import ZeroArray
  2978. m, n, k = symbols("m n k")
  2979. za = ZeroArray(m, n, k, 2)
  2980. assert _test_args(za)
  2981. def test_sympy__tensor__array__expressions__array_expressions__OneArray():
  2982. from sympy.tensor.array.expressions.array_expressions import OneArray
  2983. m, n, k = symbols("m n k")
  2984. za = OneArray(m, n, k, 2)
  2985. assert _test_args(za)
  2986. def test_sympy__tensor__functions__TensorProduct():
  2987. from sympy.tensor.functions import TensorProduct
  2988. A = MatrixSymbol('A', 3, 3)
  2989. B = MatrixSymbol('B', 3, 3)
  2990. tp = TensorProduct(A, B)
  2991. assert _test_args(tp)
  2992. def test_sympy__tensor__indexed__Idx():
  2993. from sympy.tensor.indexed import Idx
  2994. assert _test_args(Idx('test'))
  2995. assert _test_args(Idx('test', (0, 10)))
  2996. assert _test_args(Idx('test', 2))
  2997. assert _test_args(Idx('test', x))
  2998. def test_sympy__tensor__indexed__Indexed():
  2999. from sympy.tensor.indexed import Indexed, Idx
  3000. assert _test_args(Indexed('A', Idx('i'), Idx('j')))
  3001. def test_sympy__tensor__indexed__IndexedBase():
  3002. from sympy.tensor.indexed import IndexedBase
  3003. assert _test_args(IndexedBase('A', shape=(x, y)))
  3004. assert _test_args(IndexedBase('A', 1))
  3005. assert _test_args(IndexedBase('A')[0, 1])
  3006. def test_sympy__tensor__tensor__TensorIndexType():
  3007. from sympy.tensor.tensor import TensorIndexType
  3008. assert _test_args(TensorIndexType('Lorentz'))
  3009. @SKIP("deprecated class")
  3010. def test_sympy__tensor__tensor__TensorType():
  3011. pass
  3012. def test_sympy__tensor__tensor__TensorSymmetry():
  3013. from sympy.tensor.tensor import TensorSymmetry, get_symmetric_group_sgs
  3014. assert _test_args(TensorSymmetry(get_symmetric_group_sgs(2)))
  3015. def test_sympy__tensor__tensor__TensorHead():
  3016. from sympy.tensor.tensor import TensorIndexType, TensorSymmetry, get_symmetric_group_sgs, TensorHead
  3017. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3018. sym = TensorSymmetry(get_symmetric_group_sgs(1))
  3019. assert _test_args(TensorHead('p', [Lorentz], sym, 0))
  3020. def test_sympy__tensor__tensor__TensorIndex():
  3021. from sympy.tensor.tensor import TensorIndexType, TensorIndex
  3022. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3023. assert _test_args(TensorIndex('i', Lorentz))
  3024. @SKIP("abstract class")
  3025. def test_sympy__tensor__tensor__TensExpr():
  3026. pass
  3027. def test_sympy__tensor__tensor__TensAdd():
  3028. from sympy.tensor.tensor import TensorIndexType, TensorSymmetry, get_symmetric_group_sgs, tensor_indices, TensAdd, tensor_heads
  3029. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3030. a, b = tensor_indices('a,b', Lorentz)
  3031. sym = TensorSymmetry(get_symmetric_group_sgs(1))
  3032. p, q = tensor_heads('p,q', [Lorentz], sym)
  3033. t1 = p(a)
  3034. t2 = q(a)
  3035. assert _test_args(TensAdd(t1, t2))
  3036. def test_sympy__tensor__tensor__Tensor():
  3037. from sympy.tensor.tensor import TensorIndexType, TensorSymmetry, get_symmetric_group_sgs, tensor_indices, TensorHead
  3038. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3039. a, b = tensor_indices('a,b', Lorentz)
  3040. sym = TensorSymmetry(get_symmetric_group_sgs(1))
  3041. p = TensorHead('p', [Lorentz], sym)
  3042. assert _test_args(p(a))
  3043. def test_sympy__tensor__tensor__TensMul():
  3044. from sympy.tensor.tensor import TensorIndexType, TensorSymmetry, get_symmetric_group_sgs, tensor_indices, tensor_heads
  3045. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3046. a, b = tensor_indices('a,b', Lorentz)
  3047. sym = TensorSymmetry(get_symmetric_group_sgs(1))
  3048. p, q = tensor_heads('p, q', [Lorentz], sym)
  3049. assert _test_args(3*p(a)*q(b))
  3050. def test_sympy__tensor__tensor__TensorElement():
  3051. from sympy.tensor.tensor import TensorIndexType, TensorHead, TensorElement
  3052. L = TensorIndexType("L")
  3053. A = TensorHead("A", [L, L])
  3054. telem = TensorElement(A(x, y), {x: 1})
  3055. assert _test_args(telem)
  3056. def test_sympy__tensor__tensor__WildTensor():
  3057. from sympy.tensor.tensor import TensorIndexType, WildTensorHead, TensorIndex
  3058. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3059. a = TensorIndex('a', Lorentz)
  3060. p = WildTensorHead('p')
  3061. assert _test_args(p(a))
  3062. def test_sympy__tensor__tensor__WildTensorHead():
  3063. from sympy.tensor.tensor import WildTensorHead
  3064. assert _test_args(WildTensorHead('p'))
  3065. def test_sympy__tensor__tensor__WildTensorIndex():
  3066. from sympy.tensor.tensor import TensorIndexType, WildTensorIndex
  3067. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3068. assert _test_args(WildTensorIndex('i', Lorentz))
  3069. def test_sympy__tensor__toperators__PartialDerivative():
  3070. from sympy.tensor.tensor import TensorIndexType, tensor_indices, TensorHead
  3071. from sympy.tensor.toperators import PartialDerivative
  3072. Lorentz = TensorIndexType('Lorentz', dummy_name='L')
  3073. a, b = tensor_indices('a,b', Lorentz)
  3074. A = TensorHead("A", [Lorentz])
  3075. assert _test_args(PartialDerivative(A(a), A(b)))
  3076. def test_as_coeff_add():
  3077. assert (7, (3*x, 4*x**2)) == (7 + 3*x + 4*x**2).as_coeff_add()
  3078. def test_sympy__geometry__curve__Curve():
  3079. from sympy.geometry.curve import Curve
  3080. assert _test_args(Curve((x, 1), (x, 0, 1)))
  3081. def test_sympy__geometry__point__Point():
  3082. from sympy.geometry.point import Point
  3083. assert _test_args(Point(0, 1))
  3084. def test_sympy__geometry__point__Point2D():
  3085. from sympy.geometry.point import Point2D
  3086. assert _test_args(Point2D(0, 1))
  3087. def test_sympy__geometry__point__Point3D():
  3088. from sympy.geometry.point import Point3D
  3089. assert _test_args(Point3D(0, 1, 2))
  3090. def test_sympy__geometry__ellipse__Ellipse():
  3091. from sympy.geometry.ellipse import Ellipse
  3092. assert _test_args(Ellipse((0, 1), 2, 3))
  3093. def test_sympy__geometry__ellipse__Circle():
  3094. from sympy.geometry.ellipse import Circle
  3095. assert _test_args(Circle((0, 1), 2))
  3096. def test_sympy__geometry__parabola__Parabola():
  3097. from sympy.geometry.parabola import Parabola
  3098. from sympy.geometry.line import Line
  3099. assert _test_args(Parabola((0, 0), Line((2, 3), (4, 3))))
  3100. @SKIP("abstract class")
  3101. def test_sympy__geometry__line__LinearEntity():
  3102. pass
  3103. def test_sympy__geometry__line__Line():
  3104. from sympy.geometry.line import Line
  3105. assert _test_args(Line((0, 1), (2, 3)))
  3106. def test_sympy__geometry__line__Ray():
  3107. from sympy.geometry.line import Ray
  3108. assert _test_args(Ray((0, 1), (2, 3)))
  3109. def test_sympy__geometry__line__Segment():
  3110. from sympy.geometry.line import Segment
  3111. assert _test_args(Segment((0, 1), (2, 3)))
  3112. @SKIP("abstract class")
  3113. def test_sympy__geometry__line__LinearEntity2D():
  3114. pass
  3115. def test_sympy__geometry__line__Line2D():
  3116. from sympy.geometry.line import Line2D
  3117. assert _test_args(Line2D((0, 1), (2, 3)))
  3118. def test_sympy__geometry__line__Ray2D():
  3119. from sympy.geometry.line import Ray2D
  3120. assert _test_args(Ray2D((0, 1), (2, 3)))
  3121. def test_sympy__geometry__line__Segment2D():
  3122. from sympy.geometry.line import Segment2D
  3123. assert _test_args(Segment2D((0, 1), (2, 3)))
  3124. @SKIP("abstract class")
  3125. def test_sympy__geometry__line__LinearEntity3D():
  3126. pass
  3127. def test_sympy__geometry__line__Line3D():
  3128. from sympy.geometry.line import Line3D
  3129. assert _test_args(Line3D((0, 1, 1), (2, 3, 4)))
  3130. def test_sympy__geometry__line__Segment3D():
  3131. from sympy.geometry.line import Segment3D
  3132. assert _test_args(Segment3D((0, 1, 1), (2, 3, 4)))
  3133. def test_sympy__geometry__line__Ray3D():
  3134. from sympy.geometry.line import Ray3D
  3135. assert _test_args(Ray3D((0, 1, 1), (2, 3, 4)))
  3136. def test_sympy__geometry__plane__Plane():
  3137. from sympy.geometry.plane import Plane
  3138. assert _test_args(Plane((1, 1, 1), (-3, 4, -2), (1, 2, 3)))
  3139. def test_sympy__geometry__polygon__Polygon():
  3140. from sympy.geometry.polygon import Polygon
  3141. assert _test_args(Polygon((0, 1), (2, 3), (4, 5), (6, 7)))
  3142. def test_sympy__geometry__polygon__RegularPolygon():
  3143. from sympy.geometry.polygon import RegularPolygon
  3144. assert _test_args(RegularPolygon((0, 1), 2, 3, 4))
  3145. def test_sympy__geometry__polygon__Triangle():
  3146. from sympy.geometry.polygon import Triangle
  3147. assert _test_args(Triangle((0, 1), (2, 3), (4, 5)))
  3148. def test_sympy__geometry__entity__GeometryEntity():
  3149. from sympy.geometry.entity import GeometryEntity
  3150. from sympy.geometry.point import Point
  3151. assert _test_args(GeometryEntity(Point(1, 0), 1, [1, 2]))
  3152. @SKIP("abstract class")
  3153. def test_sympy__geometry__entity__GeometrySet():
  3154. pass
  3155. def test_sympy__diffgeom__diffgeom__Manifold():
  3156. from sympy.diffgeom import Manifold
  3157. assert _test_args(Manifold('name', 3))
  3158. def test_sympy__diffgeom__diffgeom__Patch():
  3159. from sympy.diffgeom import Manifold, Patch
  3160. assert _test_args(Patch('name', Manifold('name', 3)))
  3161. def test_sympy__diffgeom__diffgeom__CoordSystem():
  3162. from sympy.diffgeom import Manifold, Patch, CoordSystem
  3163. assert _test_args(CoordSystem('name', Patch('name', Manifold('name', 3))))
  3164. assert _test_args(CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c]))
  3165. def test_sympy__diffgeom__diffgeom__CoordinateSymbol():
  3166. from sympy.diffgeom import Manifold, Patch, CoordSystem, CoordinateSymbol
  3167. assert _test_args(CoordinateSymbol(CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c]), 0))
  3168. def test_sympy__diffgeom__diffgeom__Point():
  3169. from sympy.diffgeom import Manifold, Patch, CoordSystem, Point
  3170. assert _test_args(Point(
  3171. CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c]), [x, y]))
  3172. def test_sympy__diffgeom__diffgeom__BaseScalarField():
  3173. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseScalarField
  3174. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3175. assert _test_args(BaseScalarField(cs, 0))
  3176. def test_sympy__diffgeom__diffgeom__BaseVectorField():
  3177. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseVectorField
  3178. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3179. assert _test_args(BaseVectorField(cs, 0))
  3180. def test_sympy__diffgeom__diffgeom__Differential():
  3181. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseScalarField, Differential
  3182. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3183. assert _test_args(Differential(BaseScalarField(cs, 0)))
  3184. def test_sympy__diffgeom__diffgeom__Commutator():
  3185. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseVectorField, Commutator
  3186. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3187. cs1 = CoordSystem('name1', Patch('name', Manifold('name', 3)), [a, b, c])
  3188. v = BaseVectorField(cs, 0)
  3189. v1 = BaseVectorField(cs1, 0)
  3190. assert _test_args(Commutator(v, v1))
  3191. def test_sympy__diffgeom__diffgeom__TensorProduct():
  3192. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseScalarField, Differential, TensorProduct
  3193. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3194. d = Differential(BaseScalarField(cs, 0))
  3195. assert _test_args(TensorProduct(d, d))
  3196. def test_sympy__diffgeom__diffgeom__WedgeProduct():
  3197. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseScalarField, Differential, WedgeProduct
  3198. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3199. d = Differential(BaseScalarField(cs, 0))
  3200. d1 = Differential(BaseScalarField(cs, 1))
  3201. assert _test_args(WedgeProduct(d, d1))
  3202. def test_sympy__diffgeom__diffgeom__LieDerivative():
  3203. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseScalarField, Differential, BaseVectorField, LieDerivative
  3204. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3205. d = Differential(BaseScalarField(cs, 0))
  3206. v = BaseVectorField(cs, 0)
  3207. assert _test_args(LieDerivative(v, d))
  3208. def test_sympy__diffgeom__diffgeom__BaseCovarDerivativeOp():
  3209. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseCovarDerivativeOp
  3210. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3211. assert _test_args(BaseCovarDerivativeOp(cs, 0, [[[0, ]*3, ]*3, ]*3))
  3212. def test_sympy__diffgeom__diffgeom__CovarDerivativeOp():
  3213. from sympy.diffgeom import Manifold, Patch, CoordSystem, BaseVectorField, CovarDerivativeOp
  3214. cs = CoordSystem('name', Patch('name', Manifold('name', 3)), [a, b, c])
  3215. v = BaseVectorField(cs, 0)
  3216. _test_args(CovarDerivativeOp(v, [[[0, ]*3, ]*3, ]*3))
  3217. def test_sympy__categories__baseclasses__Class():
  3218. from sympy.categories.baseclasses import Class
  3219. assert _test_args(Class())
  3220. def test_sympy__categories__baseclasses__Object():
  3221. from sympy.categories import Object
  3222. assert _test_args(Object("A"))
  3223. @SKIP("abstract class")
  3224. def test_sympy__categories__baseclasses__Morphism():
  3225. pass
  3226. def test_sympy__categories__baseclasses__IdentityMorphism():
  3227. from sympy.categories import Object, IdentityMorphism
  3228. assert _test_args(IdentityMorphism(Object("A")))
  3229. def test_sympy__categories__baseclasses__NamedMorphism():
  3230. from sympy.categories import Object, NamedMorphism
  3231. assert _test_args(NamedMorphism(Object("A"), Object("B"), "f"))
  3232. def test_sympy__categories__baseclasses__CompositeMorphism():
  3233. from sympy.categories import Object, NamedMorphism, CompositeMorphism
  3234. A = Object("A")
  3235. B = Object("B")
  3236. C = Object("C")
  3237. f = NamedMorphism(A, B, "f")
  3238. g = NamedMorphism(B, C, "g")
  3239. assert _test_args(CompositeMorphism(f, g))
  3240. def test_sympy__categories__baseclasses__Diagram():
  3241. from sympy.categories import Object, NamedMorphism, Diagram
  3242. A = Object("A")
  3243. B = Object("B")
  3244. f = NamedMorphism(A, B, "f")
  3245. d = Diagram([f])
  3246. assert _test_args(d)
  3247. def test_sympy__categories__baseclasses__Category():
  3248. from sympy.categories import Object, NamedMorphism, Diagram, Category
  3249. A = Object("A")
  3250. B = Object("B")
  3251. C = Object("C")
  3252. f = NamedMorphism(A, B, "f")
  3253. g = NamedMorphism(B, C, "g")
  3254. d1 = Diagram([f, g])
  3255. d2 = Diagram([f])
  3256. K = Category("K", commutative_diagrams=[d1, d2])
  3257. assert _test_args(K)
  3258. def test_sympy__ntheory__factor___totient():
  3259. from sympy.ntheory.factor_ import totient
  3260. k = symbols('k', integer=True)
  3261. t = totient(k)
  3262. assert _test_args(t)
  3263. def test_sympy__ntheory__factor___reduced_totient():
  3264. from sympy.ntheory.factor_ import reduced_totient
  3265. k = symbols('k', integer=True)
  3266. t = reduced_totient(k)
  3267. assert _test_args(t)
  3268. def test_sympy__ntheory__factor___divisor_sigma():
  3269. from sympy.ntheory.factor_ import divisor_sigma
  3270. k = symbols('k', integer=True)
  3271. n = symbols('n', integer=True)
  3272. t = divisor_sigma(n, k)
  3273. assert _test_args(t)
  3274. def test_sympy__ntheory__factor___udivisor_sigma():
  3275. from sympy.ntheory.factor_ import udivisor_sigma
  3276. k = symbols('k', integer=True)
  3277. n = symbols('n', integer=True)
  3278. t = udivisor_sigma(n, k)
  3279. assert _test_args(t)
  3280. def test_sympy__ntheory__factor___primenu():
  3281. from sympy.ntheory.factor_ import primenu
  3282. n = symbols('n', integer=True)
  3283. t = primenu(n)
  3284. assert _test_args(t)
  3285. def test_sympy__ntheory__factor___primeomega():
  3286. from sympy.ntheory.factor_ import primeomega
  3287. n = symbols('n', integer=True)
  3288. t = primeomega(n)
  3289. assert _test_args(t)
  3290. def test_sympy__ntheory__residue_ntheory__mobius():
  3291. from sympy.ntheory import mobius
  3292. assert _test_args(mobius(2))
  3293. def test_sympy__ntheory__generate__primepi():
  3294. from sympy.ntheory import primepi
  3295. n = symbols('n')
  3296. t = primepi(n)
  3297. assert _test_args(t)
  3298. def test_sympy__physics__optics__waves__TWave():
  3299. from sympy.physics.optics import TWave
  3300. A, f, phi = symbols('A, f, phi')
  3301. assert _test_args(TWave(A, f, phi))
  3302. def test_sympy__physics__optics__gaussopt__BeamParameter():
  3303. from sympy.physics.optics import BeamParameter
  3304. assert _test_args(BeamParameter(530e-9, 1, w=1e-3, n=1))
  3305. def test_sympy__physics__optics__medium__Medium():
  3306. from sympy.physics.optics import Medium
  3307. assert _test_args(Medium('m'))
  3308. def test_sympy__physics__optics__medium__MediumN():
  3309. from sympy.physics.optics.medium import Medium
  3310. assert _test_args(Medium('m', n=2))
  3311. def test_sympy__physics__optics__medium__MediumPP():
  3312. from sympy.physics.optics.medium import Medium
  3313. assert _test_args(Medium('m', permittivity=2, permeability=2))
  3314. def test_sympy__tensor__array__expressions__array_expressions__ArrayContraction():
  3315. from sympy.tensor.array.expressions.array_expressions import ArrayContraction
  3316. from sympy.tensor.indexed import IndexedBase
  3317. A = symbols("A", cls=IndexedBase)
  3318. assert _test_args(ArrayContraction(A, (0, 1)))
  3319. def test_sympy__tensor__array__expressions__array_expressions__ArrayDiagonal():
  3320. from sympy.tensor.array.expressions.array_expressions import ArrayDiagonal
  3321. from sympy.tensor.indexed import IndexedBase
  3322. A = symbols("A", cls=IndexedBase)
  3323. assert _test_args(ArrayDiagonal(A, (0, 1)))
  3324. def test_sympy__tensor__array__expressions__array_expressions__ArrayTensorProduct():
  3325. from sympy.tensor.array.expressions.array_expressions import ArrayTensorProduct
  3326. from sympy.tensor.indexed import IndexedBase
  3327. A, B = symbols("A B", cls=IndexedBase)
  3328. assert _test_args(ArrayTensorProduct(A, B))
  3329. def test_sympy__tensor__array__expressions__array_expressions__ArrayAdd():
  3330. from sympy.tensor.array.expressions.array_expressions import ArrayAdd
  3331. from sympy.tensor.indexed import IndexedBase
  3332. A, B = symbols("A B", cls=IndexedBase)
  3333. assert _test_args(ArrayAdd(A, B))
  3334. def test_sympy__tensor__array__expressions__array_expressions__PermuteDims():
  3335. from sympy.tensor.array.expressions.array_expressions import PermuteDims
  3336. A = MatrixSymbol("A", 4, 4)
  3337. assert _test_args(PermuteDims(A, (1, 0)))
  3338. def test_sympy__tensor__array__expressions__array_expressions__ArrayElementwiseApplyFunc():
  3339. from sympy.tensor.array.expressions.array_expressions import ArraySymbol, ArrayElementwiseApplyFunc
  3340. A = ArraySymbol("A", (4,))
  3341. assert _test_args(ArrayElementwiseApplyFunc(exp, A))
  3342. def test_sympy__tensor__array__expressions__array_expressions__Reshape():
  3343. from sympy.tensor.array.expressions.array_expressions import ArraySymbol, Reshape
  3344. A = ArraySymbol("A", (4,))
  3345. assert _test_args(Reshape(A, (2, 2)))
  3346. def test_sympy__codegen__ast__Assignment():
  3347. from sympy.codegen.ast import Assignment
  3348. assert _test_args(Assignment(x, y))
  3349. def test_sympy__codegen__cfunctions__expm1():
  3350. from sympy.codegen.cfunctions import expm1
  3351. assert _test_args(expm1(x))
  3352. def test_sympy__codegen__cfunctions__log1p():
  3353. from sympy.codegen.cfunctions import log1p
  3354. assert _test_args(log1p(x))
  3355. def test_sympy__codegen__cfunctions__exp2():
  3356. from sympy.codegen.cfunctions import exp2
  3357. assert _test_args(exp2(x))
  3358. def test_sympy__codegen__cfunctions__log2():
  3359. from sympy.codegen.cfunctions import log2
  3360. assert _test_args(log2(x))
  3361. def test_sympy__codegen__cfunctions__fma():
  3362. from sympy.codegen.cfunctions import fma
  3363. assert _test_args(fma(x, y, z))
  3364. def test_sympy__codegen__cfunctions__log10():
  3365. from sympy.codegen.cfunctions import log10
  3366. assert _test_args(log10(x))
  3367. def test_sympy__codegen__cfunctions__Sqrt():
  3368. from sympy.codegen.cfunctions import Sqrt
  3369. assert _test_args(Sqrt(x))
  3370. def test_sympy__codegen__cfunctions__Cbrt():
  3371. from sympy.codegen.cfunctions import Cbrt
  3372. assert _test_args(Cbrt(x))
  3373. def test_sympy__codegen__cfunctions__hypot():
  3374. from sympy.codegen.cfunctions import hypot
  3375. assert _test_args(hypot(x, y))
  3376. def test_sympy__codegen__fnodes__FFunction():
  3377. from sympy.codegen.fnodes import FFunction
  3378. assert _test_args(FFunction('f'))
  3379. def test_sympy__codegen__fnodes__F95Function():
  3380. from sympy.codegen.fnodes import F95Function
  3381. assert _test_args(F95Function('f'))
  3382. def test_sympy__codegen__fnodes__isign():
  3383. from sympy.codegen.fnodes import isign
  3384. assert _test_args(isign(1, x))
  3385. def test_sympy__codegen__fnodes__dsign():
  3386. from sympy.codegen.fnodes import dsign
  3387. assert _test_args(dsign(1, x))
  3388. def test_sympy__codegen__fnodes__cmplx():
  3389. from sympy.codegen.fnodes import cmplx
  3390. assert _test_args(cmplx(x, y))
  3391. def test_sympy__codegen__fnodes__kind():
  3392. from sympy.codegen.fnodes import kind
  3393. assert _test_args(kind(x))
  3394. def test_sympy__codegen__fnodes__merge():
  3395. from sympy.codegen.fnodes import merge
  3396. assert _test_args(merge(1, 2, Eq(x, 0)))
  3397. def test_sympy__codegen__fnodes___literal():
  3398. from sympy.codegen.fnodes import _literal
  3399. assert _test_args(_literal(1))
  3400. def test_sympy__codegen__fnodes__literal_sp():
  3401. from sympy.codegen.fnodes import literal_sp
  3402. assert _test_args(literal_sp(1))
  3403. def test_sympy__codegen__fnodes__literal_dp():
  3404. from sympy.codegen.fnodes import literal_dp
  3405. assert _test_args(literal_dp(1))
  3406. def test_sympy__codegen__matrix_nodes__MatrixSolve():
  3407. from sympy.matrices import MatrixSymbol
  3408. from sympy.codegen.matrix_nodes import MatrixSolve
  3409. A = MatrixSymbol('A', 3, 3)
  3410. v = MatrixSymbol('x', 3, 1)
  3411. assert _test_args(MatrixSolve(A, v))
  3412. def test_sympy__vector__coordsysrect__CoordSys3D():
  3413. from sympy.vector.coordsysrect import CoordSys3D
  3414. assert _test_args(CoordSys3D('C'))
  3415. def test_sympy__vector__point__Point():
  3416. from sympy.vector.point import Point
  3417. assert _test_args(Point('P'))
  3418. def test_sympy__vector__basisdependent__BasisDependent():
  3419. #from sympy.vector.basisdependent import BasisDependent
  3420. #These classes have been created to maintain an OOP hierarchy
  3421. #for Vectors and Dyadics. Are NOT meant to be initialized
  3422. pass
  3423. def test_sympy__vector__basisdependent__BasisDependentMul():
  3424. #from sympy.vector.basisdependent import BasisDependentMul
  3425. #These classes have been created to maintain an OOP hierarchy
  3426. #for Vectors and Dyadics. Are NOT meant to be initialized
  3427. pass
  3428. def test_sympy__vector__basisdependent__BasisDependentAdd():
  3429. #from sympy.vector.basisdependent import BasisDependentAdd
  3430. #These classes have been created to maintain an OOP hierarchy
  3431. #for Vectors and Dyadics. Are NOT meant to be initialized
  3432. pass
  3433. def test_sympy__vector__basisdependent__BasisDependentZero():
  3434. #from sympy.vector.basisdependent import BasisDependentZero
  3435. #These classes have been created to maintain an OOP hierarchy
  3436. #for Vectors and Dyadics. Are NOT meant to be initialized
  3437. pass
  3438. def test_sympy__vector__vector__BaseVector():
  3439. from sympy.vector.vector import BaseVector
  3440. from sympy.vector.coordsysrect import CoordSys3D
  3441. C = CoordSys3D('C')
  3442. assert _test_args(BaseVector(0, C, ' ', ' '))
  3443. def test_sympy__vector__vector__VectorAdd():
  3444. from sympy.vector.vector import VectorAdd, VectorMul
  3445. from sympy.vector.coordsysrect import CoordSys3D
  3446. C = CoordSys3D('C')
  3447. from sympy.abc import a, b, c, x, y, z
  3448. v1 = a*C.i + b*C.j + c*C.k
  3449. v2 = x*C.i + y*C.j + z*C.k
  3450. assert _test_args(VectorAdd(v1, v2))
  3451. assert _test_args(VectorMul(x, v1))
  3452. def test_sympy__vector__vector__VectorMul():
  3453. from sympy.vector.vector import VectorMul
  3454. from sympy.vector.coordsysrect import CoordSys3D
  3455. C = CoordSys3D('C')
  3456. from sympy.abc import a
  3457. assert _test_args(VectorMul(a, C.i))
  3458. def test_sympy__vector__vector__VectorZero():
  3459. from sympy.vector.vector import VectorZero
  3460. assert _test_args(VectorZero())
  3461. def test_sympy__vector__vector__Vector():
  3462. #from sympy.vector.vector import Vector
  3463. #Vector is never to be initialized using args
  3464. pass
  3465. def test_sympy__vector__vector__Cross():
  3466. from sympy.vector.vector import Cross
  3467. from sympy.vector.coordsysrect import CoordSys3D
  3468. C = CoordSys3D('C')
  3469. _test_args(Cross(C.i, C.j))
  3470. def test_sympy__vector__vector__Dot():
  3471. from sympy.vector.vector import Dot
  3472. from sympy.vector.coordsysrect import CoordSys3D
  3473. C = CoordSys3D('C')
  3474. _test_args(Dot(C.i, C.j))
  3475. def test_sympy__vector__dyadic__Dyadic():
  3476. #from sympy.vector.dyadic import Dyadic
  3477. #Dyadic is never to be initialized using args
  3478. pass
  3479. def test_sympy__vector__dyadic__BaseDyadic():
  3480. from sympy.vector.dyadic import BaseDyadic
  3481. from sympy.vector.coordsysrect import CoordSys3D
  3482. C = CoordSys3D('C')
  3483. assert _test_args(BaseDyadic(C.i, C.j))
  3484. def test_sympy__vector__dyadic__DyadicMul():
  3485. from sympy.vector.dyadic import BaseDyadic, DyadicMul
  3486. from sympy.vector.coordsysrect import CoordSys3D
  3487. C = CoordSys3D('C')
  3488. assert _test_args(DyadicMul(3, BaseDyadic(C.i, C.j)))
  3489. def test_sympy__vector__dyadic__DyadicAdd():
  3490. from sympy.vector.dyadic import BaseDyadic, DyadicAdd
  3491. from sympy.vector.coordsysrect import CoordSys3D
  3492. C = CoordSys3D('C')
  3493. assert _test_args(2 * DyadicAdd(BaseDyadic(C.i, C.i),
  3494. BaseDyadic(C.i, C.j)))
  3495. def test_sympy__vector__dyadic__DyadicZero():
  3496. from sympy.vector.dyadic import DyadicZero
  3497. assert _test_args(DyadicZero())
  3498. def test_sympy__vector__deloperator__Del():
  3499. from sympy.vector.deloperator import Del
  3500. assert _test_args(Del())
  3501. def test_sympy__vector__implicitregion__ImplicitRegion():
  3502. from sympy.vector.implicitregion import ImplicitRegion
  3503. from sympy.abc import x, y
  3504. assert _test_args(ImplicitRegion((x, y), y**3 - 4*x))
  3505. def test_sympy__vector__integrals__ParametricIntegral():
  3506. from sympy.vector.integrals import ParametricIntegral
  3507. from sympy.vector.parametricregion import ParametricRegion
  3508. from sympy.vector.coordsysrect import CoordSys3D
  3509. C = CoordSys3D('C')
  3510. assert _test_args(ParametricIntegral(C.y*C.i - 10*C.j,\
  3511. ParametricRegion((x, y), (x, 1, 3), (y, -2, 2))))
  3512. def test_sympy__vector__operators__Curl():
  3513. from sympy.vector.operators import Curl
  3514. from sympy.vector.coordsysrect import CoordSys3D
  3515. C = CoordSys3D('C')
  3516. assert _test_args(Curl(C.i))
  3517. def test_sympy__vector__operators__Laplacian():
  3518. from sympy.vector.operators import Laplacian
  3519. from sympy.vector.coordsysrect import CoordSys3D
  3520. C = CoordSys3D('C')
  3521. assert _test_args(Laplacian(C.i))
  3522. def test_sympy__vector__operators__Divergence():
  3523. from sympy.vector.operators import Divergence
  3524. from sympy.vector.coordsysrect import CoordSys3D
  3525. C = CoordSys3D('C')
  3526. assert _test_args(Divergence(C.i))
  3527. def test_sympy__vector__operators__Gradient():
  3528. from sympy.vector.operators import Gradient
  3529. from sympy.vector.coordsysrect import CoordSys3D
  3530. C = CoordSys3D('C')
  3531. assert _test_args(Gradient(C.x))
  3532. def test_sympy__vector__orienters__Orienter():
  3533. #from sympy.vector.orienters import Orienter
  3534. #Not to be initialized
  3535. pass
  3536. def test_sympy__vector__orienters__ThreeAngleOrienter():
  3537. #from sympy.vector.orienters import ThreeAngleOrienter
  3538. #Not to be initialized
  3539. pass
  3540. def test_sympy__vector__orienters__AxisOrienter():
  3541. from sympy.vector.orienters import AxisOrienter
  3542. from sympy.vector.coordsysrect import CoordSys3D
  3543. C = CoordSys3D('C')
  3544. assert _test_args(AxisOrienter(x, C.i))
  3545. def test_sympy__vector__orienters__BodyOrienter():
  3546. from sympy.vector.orienters import BodyOrienter
  3547. assert _test_args(BodyOrienter(x, y, z, '123'))
  3548. def test_sympy__vector__orienters__SpaceOrienter():
  3549. from sympy.vector.orienters import SpaceOrienter
  3550. assert _test_args(SpaceOrienter(x, y, z, '123'))
  3551. def test_sympy__vector__orienters__QuaternionOrienter():
  3552. from sympy.vector.orienters import QuaternionOrienter
  3553. a, b, c, d = symbols('a b c d')
  3554. assert _test_args(QuaternionOrienter(a, b, c, d))
  3555. def test_sympy__vector__parametricregion__ParametricRegion():
  3556. from sympy.abc import t
  3557. from sympy.vector.parametricregion import ParametricRegion
  3558. assert _test_args(ParametricRegion((t, t**3), (t, 0, 2)))
  3559. def test_sympy__vector__scalar__BaseScalar():
  3560. from sympy.vector.scalar import BaseScalar
  3561. from sympy.vector.coordsysrect import CoordSys3D
  3562. C = CoordSys3D('C')
  3563. assert _test_args(BaseScalar(0, C, ' ', ' '))
  3564. def test_sympy__physics__wigner__Wigner3j():
  3565. from sympy.physics.wigner import Wigner3j
  3566. assert _test_args(Wigner3j(0, 0, 0, 0, 0, 0))
  3567. def test_sympy__combinatorics__schur_number__SchurNumber():
  3568. from sympy.combinatorics.schur_number import SchurNumber
  3569. assert _test_args(SchurNumber(x))
  3570. def test_sympy__combinatorics__perm_groups__SymmetricPermutationGroup():
  3571. from sympy.combinatorics.perm_groups import SymmetricPermutationGroup
  3572. assert _test_args(SymmetricPermutationGroup(5))
  3573. def test_sympy__combinatorics__perm_groups__Coset():
  3574. from sympy.combinatorics.permutations import Permutation
  3575. from sympy.combinatorics.perm_groups import PermutationGroup, Coset
  3576. a = Permutation(1, 2)
  3577. b = Permutation(0, 1)
  3578. G = PermutationGroup([a, b])
  3579. assert _test_args(Coset(a, G))