def test_serial_sum_dict(self): msngr = messenger.Messenger() data = {'a': 1, 'b': 2, 'c': 3} msngr_sum = msngr.sum(data) print msngr_sum self.assertDictEqual(msngr_sum, data, 'Serial messenger dict sum not working')
def test_serial_max_dict(self): msngr = messenger.Messenger() data = {'a': 1, 'b': 2, 'c': 3} msngr_max = msngr.sum(data) print msngr_max self.assertDictEqual(msngr_max, data, 'Serial messenger dict max not working')
def test_serial_init(self): msngr = messenger.Messenger() self.assertIsInstance(msngr, messenger.Messenger, 'Failed to create class instance') self.assertEqual(msngr._mpi_rank, 0, 'Rank is wrong after initialization') self.assertEqual(msngr._mpi_size, 1, 'Size is wrong after initialization') self.assertEqual(msngr._is_master, (0 == 0), 'Is_master is wrong after initialization') self.assertEqual(msngr.verbosity, 1, 'Verbosity is wrong after initialization')
def test_serial_print_all(self): msngr = messenger.Messenger() msg = 'TEST - ALL - SERIAL' msngr.print_all(msg, vlevel=0)
def test_serial_print_once(self): msngr = messenger.Messenger() msg = 'TEST - ONCE - SERIAL' msngr.print_once(msg, vlevel=0)
def test_serial_max_list(self): msngr = messenger.Messenger() data = [1, 2, 3] msngr_max = msngr.sum(data) print msngr_max self.assertEqual(msngr_max, 6, 'Serial messenger list max not working')
def test_serial_sum_list(self): msngr = messenger.Messenger() data = [1, 2, 3] msngr_sum = msngr.sum(data) print msngr_sum self.assertEqual(msngr_sum, 6, 'Serial messenger list sum not working')
def test_serial_get_size(self): msngr = messenger.Messenger() self.assertEqual(msngr.get_size(), 1, 'Serial messenger size should be 1')
def test_serial_get_rank(self): msngr = messenger.Messenger() self.assertEqual(msngr.get_rank(), 0, 'Serial messenger rank should be 0')
def test_serial_is_master(self): msngr = messenger.Messenger() self.assertTrue(msngr.is_master(), 'Serial messenger should be master')
def test_serial_partition(self): msngr = messenger.Messenger() data = [1, 2, 3] p_data = msngr.partition(data) self.assertListEqual(data, p_data, 'Serial partition is wrong')