test_math_02.py

from math import (sin, cos, tan, pi, sqrt, log, log10, log2, erf, erfc, gamma,
                  lgamma, asin, acos, atan, atan2, asinh, acosh, atanh,
                  tanh, sinh, cosh, hypot, copysign)
from lpython import f64

def test_trig():
    eps: f64 = 1e-12
    assert abs(sin(0.0) - 0.0) < eps
    assert abs(sin(pi/2.0) - 1.0) < eps
    assert abs(cos(0.0) - 1.0) < eps
    assert abs(cos(pi/2.0) - 0.0) < eps
    assert abs(tan(0.0) - 0.0) < eps
    assert abs(tan(pi/4.0) - 1.0) < eps
    assert abs(asin(1.0) - pi/2.0) < eps
    assert abs(acos(1.0) - 0.0) < eps
    assert abs(atan(1.0) - pi/4.0) < eps
    assert abs(atan2(1.0, 1.0) - pi/4.0) < eps

def test_sqrt():
    eps: f64 = 1e-12
    assert abs(sqrt(2.0) - 1.4142135623730951) < eps
    assert abs(sqrt(9.0) - 3.0) < eps

def test_log():
    eps: f64 = 1e-12
    assert abs(log(1.0) - 0.0) < eps
    assert abs(log(2.718281828459) - 1.0) < eps
    assert abs(log2(2.0) - 1.0) < eps
    assert abs(log10(10.0) - 1.0) < eps

def test_special():
    eps: f64 = 1e-12
    assert abs(erf(2.0) - 0.9953222650189527) < eps
    assert abs(erfc(2.0) - 0.0046777349810472645) < eps
    assert abs(erf(2.0) + erfc(2.0) - 1.0) < eps
    assert abs(gamma(5.0) - 24.0) < eps
    assert abs(lgamma(5.0) - log(gamma(5.0))) < eps

def test_hyperbolic():
    eps: f64 = 1e-12
    assert abs(sinh(1.0) - 1.1752011936438014) < eps
    assert abs(cosh(1.0) - 1.5430806348152437) < eps
    assert abs(tanh(1.0) - 0.7615941559557649) < eps
    assert abs(asinh(1.0) - 0.881373587019543) < eps

def test_copysign():
    eps: f64 = 1e-12
    assert abs(copysign(-8.56, 97.21) - 8.56) < eps
    assert abs(copysign(-43.0, -76.67) - (-43.0)) < eps

def test_hypot():
    eps: f64 = 1e-12
    assert abs(hypot(3, 4) - 5.0) < eps
    assert abs(hypot(-3, 4) - 5.0) < eps
    assert abs(hypot(6, 6) - 8.48528137423857) < eps

def check():
    test_trig()
    test_sqrt()
    test_log()
    test_special()
    test_hyperbolic()
    test_copysign()
    test_hypot()

check()