#!/usr/bin/env python3

import unittest
import os
import numpy as np
from booz_xform import Booz_xform

TEST_DIR = os.path.join(os.path.dirname(__file__), 'test_files')

class MainTest(unittest.TestCase):
    def test_compute_surfs_edit(self):
        b = Booz_xform()
        b.read_wout(os.path.join(TEST_DIR, 'wout_li383_1.4m.nc'))
        b.compute_surfs = [10, 15]
        np.testing.assert_allclose(b.compute_surfs, [10, 15])

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

#!/usr/bin/env python3

import unittest
import os
import numpy as np
from booz_xform import Booz_xform

TEST_DIR = os.path.join(os.path.dirname(__file__), 'test_files')

class MainTest(unittest.TestCase):
    def test_compute_surfs_edit(self):
        b = Booz_xform()
        b.read_wout(os.path.join(TEST_DIR, 'wout_li383_1.4m.nc'))
        b.compute_surfs = [10, 15]
        np.testing.assert_allclose(b.compute_surfs, [10, 15])

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

#!/usr/bin/env python3