test/hoomd_script/test_update_zero_momentum.py

   1 # -*- coding: iso-8859-1 -*-
   2 # Maintainer: joaander
   3
   4 from hoomd_script import *
   5 import unittest
   6 import os
   7
   8 # tests for update.zero_momentum
   9 class update_zero_momentum_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 updater
  17     def test(self):
  18         update.zero_momentum()
  19         run(100);
  20
  21     # test variable periods
  22     def test_variable(self):
  23         update.zero_momentum(period = lambda n: n*100);
  24         run(100);
  25
  26     def tearDown(self):
  27         init.reset();
  28
  29 if __name__ == '__main__':
  30     unittest.main(argv = ['test.py', '-v'])