| from mpmath import * |
| from mpmath.libmp import ifac |
|
|
| import sys |
| if "-dps" in sys.argv: |
| maxdps = int(sys.argv[sys.argv.index("-dps")+1]) |
| else: |
| maxdps = 1000 |
|
|
| raise_ = "-raise" in sys.argv |
|
|
| errcount = 0 |
|
|
| def check(name, func, z, y): |
| global errcount |
| try: |
| x = func(z) |
| except: |
| errcount += 1 |
| if raise_: |
| raise |
| print() |
| print(name) |
| print("EXCEPTION") |
| import traceback |
| traceback.print_tb(sys.exc_info()[2]) |
| print() |
| return |
| xre = x.real |
| xim = x.imag |
| yre = y.real |
| yim = y.imag |
| tol = eps*8 |
| err = 0 |
| if abs(xre-yre) > abs(yre)*tol: |
| err = 1 |
| print() |
| print("Error! %s (re = %s, wanted %s, err=%s)" % (name, nstr(xre,10), nstr(yre,10), nstr(abs(xre-yre)))) |
| errcount += 1 |
| if raise_: |
| raise SystemExit |
| if abs(xim-yim) > abs(yim)*tol: |
| err = 1 |
| print() |
| print("Error! %s (im = %s, wanted %s, err=%s)" % (name, nstr(xim,10), nstr(yim,10), nstr(abs(xim-yim)))) |
| errcount += 1 |
| if raise_: |
| raise SystemExit |
| if not err: |
| sys.stdout.write("%s ok; " % name) |
|
|
| def testcase(case): |
| z, result = case |
| print("Testing z =", z) |
| mp.dps = 1010 |
| z = eval(z) |
| mp.dps = maxdps + 50 |
| if result is None: |
| gamma_val = gamma(z) |
| loggamma_val = loggamma(z) |
| factorial_val = factorial(z) |
| rgamma_val = rgamma(z) |
| else: |
| loggamma_val = eval(result) |
| gamma_val = exp(loggamma_val) |
| factorial_val = z * gamma_val |
| rgamma_val = 1/gamma_val |
| for dps in [5, 10, 15, 25, 40, 60, 90, 120, 250, 600, 1000, 1800, 3600]: |
| if dps > maxdps: |
| break |
| mp.dps = dps |
| print("dps = %s" % dps) |
| check("gamma", gamma, z, gamma_val) |
| check("rgamma", rgamma, z, rgamma_val) |
| check("loggamma", loggamma, z, loggamma_val) |
| check("factorial", factorial, z, factorial_val) |
| print() |
| mp.dps = 15 |
|
|
| testcases = [] |
|
|
| |
| for n in list(range(1,200)) + list(range(201,2000,17)): |
| testcases.append(["%s" % n, None]) |
| for n in range(-200,200): |
| testcases.append(["%s+0.5" % n, None]) |
| testcases.append(["%s+0.37" % n, None]) |
|
|
| testcases += [\ |
| ["(0.1+1j)", None], |
| ["(-0.1+1j)", None], |
| ["(0.1-1j)", None], |
| ["(-0.1-1j)", None], |
| ["10j", None], |
| ["-10j", None], |
| ["100j", None], |
| ["10000j", None], |
| ["-10000000j", None], |
| ["(10**100)*j", None], |
| ["125+(10**100)*j", None], |
| ["-125+(10**100)*j", None], |
| ["(10**10)*(1+j)", None], |
| ["(10**10)*(-1+j)", None], |
| ["(10**100)*(1+j)", None], |
| ["(10**100)*(-1+j)", None], |
| ["(1.5-1j)", None], |
| ["(6+4j)", None], |
| ["(4+1j)", None], |
| ["(3.5+2j)", None], |
| ["(1.5-1j)", None], |
| ["(-6-4j)", None], |
| ["(-2-3j)", None], |
| ["(-2.5-2j)", None], |
| ["(4+1j)", None], |
| ["(3+3j)", None], |
| ["(2-2j)", None], |
| ["1", "0"], |
| ["2", "0"], |
| ["3", "log(2)"], |
| ["4", "log(6)"], |
| ["5", "log(24)"], |
| ["0.5", "log(pi)/2"], |
| ["1.5", "log(sqrt(pi)/2)"], |
| ["2.5", "log(3*sqrt(pi)/4)"], |
| ["mpf('0.37')", None], |
| ["0.25", "log(sqrt(2*sqrt(2*pi**3)/agm(1,sqrt(2))))"], |
| ["-0.4", None], |
| ["mpf('-1.9')", None], |
| ["mpf('12.8')", None], |
| ["mpf('33.7')", None], |
| ["mpf('95.2')", None], |
| ["mpf('160.3')", None], |
| ["mpf('2057.8')", None], |
| ["25", "log(ifac(24))"], |
| ["80", "log(ifac(79))"], |
| ["500", "log(ifac(500-1))"], |
| ["8000", "log(ifac(8000-1))"], |
| ["8000.5", None], |
| ["mpf('8000.1')", None], |
| ["mpf('1.37e10')", None], |
| ["mpf('1.37e10')*(1+j)", None], |
| ["mpf('1.37e10')*(-1+j)", None], |
| ["mpf('1.37e10')*(-1-j)", None], |
| ["mpf('1.37e10')*(-1+j)", None], |
| ["mpf('1.37e100')", None], |
| ["mpf('1.37e100')*(1+j)", None], |
| ["mpf('1.37e100')*(-1+j)", None], |
| ["mpf('1.37e100')*(-1-j)", None], |
| ["mpf('1.37e100')*(-1+j)", None], |
| ["3+4j", |
| "mpc('" |
| "-1.7566267846037841105306041816232757851567066070613445016197619371316057169" |
| "4723618263960834804618463052988607348289672535780644470689771115236512106002" |
| "5970873471563240537307638968509556191696167970488390423963867031934333890838" |
| "8009531786948197210025029725361069435208930363494971027388382086721660805397" |
| "9163230643216054580167976201709951509519218635460317367338612500626714783631" |
| "7498317478048447525674016344322545858832610325861086336204591943822302971823" |
| "5161814175530618223688296232894588415495615809337292518431903058265147109853" |
| "1710568942184987827643886816200452860853873815413367529829631430146227470517" |
| "6579967222200868632179482214312673161276976117132204633283806161971389519137" |
| "1243359764435612951384238091232760634271570950240717650166551484551654327989" |
| "9360285030081716934130446150245110557038117075172576825490035434069388648124" |
| "6678152254554001586736120762641422590778766100376515737713938521275749049949" |
| "1284143906816424244705094759339932733567910991920631339597278805393743140853" |
| "391550313363278558195609260225928','" |
| "4.74266443803465792819488940755002274088830335171164611359052405215840070271" |
| "5906813009373171139767051863542508136875688550817670379002790304870822775498" |
| "2809996675877564504192565392367259119610438951593128982646945990372179860613" |
| "4294436498090428077839141927485901735557543641049637962003652638924845391650" |
| "9546290137755550107224907606529385248390667634297183361902055842228798984200" |
| "9591180450211798341715874477629099687609819466457990642030707080894518168924" |
| "6805549314043258530272479246115112769957368212585759640878745385160943755234" |
| "9398036774908108204370323896757543121853650025529763655312360354244898913463" |
| "7115955702828838923393113618205074162812089732064414530813087483533203244056" |
| "0546577484241423134079056537777170351934430586103623577814746004431994179990" |
| "5318522939077992613855205801498201930221975721246498720895122345420698451980" |
| "0051215797310305885845964334761831751370672996984756815410977750799748813563" |
| "8784405288158432214886648743541773208808731479748217023665577802702269468013" |
| "673719173759245720489020315779001')"], |
| ] |
|
|
| for z in [4, 14, 34, 64]: |
| testcases.append(["(2+j)*%s/3" % z, None]) |
| testcases.append(["(-2+j)*%s/3" % z, None]) |
| testcases.append(["(1+2*j)*%s/3" % z, None]) |
| testcases.append(["(2-j)*%s/3" % z, None]) |
| testcases.append(["(20+j)*%s/3" % z, None]) |
| testcases.append(["(-20+j)*%s/3" % z, None]) |
| testcases.append(["(1+20*j)*%s/3" % z, None]) |
| testcases.append(["(20-j)*%s/3" % z, None]) |
| testcases.append(["(200+j)*%s/3" % z, None]) |
| testcases.append(["(-200+j)*%s/3" % z, None]) |
| testcases.append(["(1+200*j)*%s/3" % z, None]) |
| testcases.append(["(200-j)*%s/3" % z, None]) |
|
|
| |
| for n in [0,1,2,3,4,25,-1,-2,-3,-4,-20,-21,-50,-51,-200,-201,-20000,-20001]: |
| for t in ['1e-5', '1e-20', '1e-100', '1e-10000']: |
| testcases.append(["fadd(%s,'%s',exact=True)" % (n, t), None]) |
| testcases.append(["fsub(%s,'%s',exact=True)" % (n, t), None]) |
| testcases.append(["fadd(%s,'%sj',exact=True)" % (n, t), None]) |
| testcases.append(["fsub(%s,'%sj',exact=True)" % (n, t), None]) |
|
|
| if __name__ == "__main__": |
| from timeit import default_timer as clock |
| tot_time = 0.0 |
| for case in testcases: |
| t1 = clock() |
| testcase(case) |
| t2 = clock() |
| print("Test time:", t2-t1) |
| print() |
| tot_time += (t2-t1) |
| print("Total time:", tot_time) |
| print("Errors:", errcount) |
|
|
