Test_dacf.py 877 Bytes
Newer Older
1
2
#!/usr/bin/python

3
import unittest
4
from Tests.UnitTests.UnitTest import UnitTest
5
6
from MDANSE import REGISTRY

7
class TestDACF(UnitTest):
8

9
10
11
12
13
    def test(self):
        parameters = {}
        parameters['atom_charges'] = {0: 0.5, 1: 1.2, 2: -0.2}
        parameters['atom_selection'] = 'atom_index 0,1,2'
        parameters['frames'] = (0, 10, 1)
14
        parameters['output_files'] = ('/tmp/output', ['netcdf'])
15
16
        parameters['running_mode'] = ('monoprocessor', 1)
        parameters['trajectory'] = '../../../Data/Trajectories/MMTK/waterbox_in_periodic_universe.nc'
17
18
        job = REGISTRY['job']['dacf']()
        self.assertNotRaises(job.run, parameters, status=False)
19

20
21
22
23
24
def suite():
    loader = unittest.TestLoader()
    s = unittest.TestSuite()
    s.addTest(loader.loadTestsFromTestCase(TestDACF))
    return s
25

26
27
if __name__ == '__main__':
    unittest.main(verbosity=2)