L2G/tests/test_module.py at main · iterorganization/L2G · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from tabnanny import verbose
import unittest

from l2g.external.tlas import PyTLAS
from l2g.external.flt import PyFLT
from l2g.external.equilibrium_analysis import EQA
from l2g.external.bicubic import PyBicubic
from l2g.external.bfgs_2d import PyBfgs2d
from l2g.external.rkf45 import PyRKF45FLT
import numpy as np


class TestModule(unittest.TestCase):

    def test_flt_object_creation(self):
        for _ in range(10):
            obj = PyFLT()

    def test_tlas_object_creation(self):
        for _ in range(10):
            obj = PyTLAS()

    def test_assign_tlas_to_flt(self):
        for _ in range(10):
            tlas_obj = PyTLAS()
            flt_obj = PyFLT()
            flt_obj.applyRT(tlas_obj)


    def test_PyBfgs2d_object_creation(self):
        for _ in range(10):
            obj = PyBfgs2d()
            del obj

    def test_PyRKF45FLT_object_creation(self):
        for _ in range(10):
            obj = PyRKF45FLT()
            del obj

    def test_EQA_object_creation(self):
        for _ in range(10):
            obj = EQA()
            del obj

if __name__ == '__main__':
    unittest.main()