test/hoomd_script/test_analyze_msd.py

   1 # -*- coding: iso-8859-1 -*-
   2 # Maintainer: joaander
   3
   4 from hoomd_script import *
   5 import unittest
   6 import os
   7
   8 # unit tests for analyze.msd
   9 class analyze_msd_tests (unittest.TestCase):
  10     def setUp(self):
  11         print
  12         init.create_random(N=100, phi_p=0.05);
  13
  14         sorter.set_params(grid=8)
  15
  16     # tests basic creation of the analyzer
  17     def test(self):
  18         analyze.msd(period = 10, filename="test.log", groups=[group.all()]);
  19         run(100);
  20
  21     # test variable period
  22     def test_variable(self):
  23         analyze.msd(period = lambda n: n*10, filename="test.log", groups=[group.all()]);
  24         run(100);
  25
  26     # test error if no groups defined
  27     def test_no_gropus(self):
  28         self.assertRaises(RuntimeError, analyze.msd, period=10, filename="test.log", groups=[]);
  29
  30     # test set_params
  31     def test_set_params(self):
  32         ana = analyze.msd(period = 10, filename="test.log", groups=[group.all()]);
  33         ana.set_params(delimiter = ' ');
  34         run(100);
  35
  36     def tearDown(self):
  37         init.reset();
  38         if comm.get_rank() == 0:
  39             os.remove("test.log");
  40
  41 if __name__ == '__main__':
  42     unittest.main(argv = ['test.py', '-v'])