Source Code for Module solvcon.tests.test_hook

 1  from unittest import TestCase 
 2   
 3 -class TestHook(TestCase): 
 4 -    def test_existence(self): 
 5          from .. import hook 
 6          self.assertTrue(hook.Hook) 
 7          self.assertTrue(hook.ProgressHook) 
 8   
 9 -    def test_type(self): 
10          from .. import hook 
11          self.assertRaises(AssertionError, hook.Hook, None) 
12   
13 -    def test_hookmethods(self): 
14          from ..hook import Hook 
15          self.assertTrue(getattr(Hook, 'preloop', False)) 
16          self.assertTrue(getattr(Hook, 'premarch', False)) 
17          self.assertTrue(getattr(Hook, 'postmarch', False)) 
18          self.assertTrue(getattr(Hook, 'postloop', False)) 
19   
20 -    def test_progress(self): 
21          from ..hook import Hook, ProgressHook 
22          self.assertNotEqual(ProgressHook.preloop, Hook.preloop) 
23          self.assertEqual(ProgressHook.premarch, Hook.premarch) 
24          self.assertNotEqual(ProgressHook.postmarch, Hook.postmarch) 
25          self.assertEqual(ProgressHook.postloop, Hook.postloop) 
26   
27 -class TestBlockHook(TestCase): 
28 -    def test_type(self): 
29          from .. import hook 
30          self.assertRaises(AssertionError, hook.BlockHook, None) 
31   
32 -    def test_blockhook(self): 
33          from ..hook import Hook, BlockHook 
34          self.assertEqual(BlockHook.preloop, Hook.preloop) 
35          self.assertEqual(BlockHook.premarch, Hook.premarch) 
36          self.assertEqual(BlockHook.postmarch, Hook.postmarch) 
37          self.assertEqual(BlockHook.postloop, Hook.postloop) 
38   
39 -    def test_blockinfohook(self): 
40          from ..hook import Hook, BlockInfoHook 
41          self.assertNotEqual(BlockInfoHook.preloop, Hook.preloop) 
42          self.assertEqual(BlockInfoHook.premarch, Hook.premarch) 
43          self.assertEqual(BlockInfoHook.postmarch, Hook.postmarch) 
44          self.assertNotEqual(BlockInfoHook.postloop, Hook.postloop) 
45   
46 -    def test_vtksave(self): 
47          from ..case import BlockCase 
48          from ..hook import Hook, VtkSave 
49          cse = BlockCase() 
50          hok = VtkSave(cse) 
51          self.assertEqual(hok.binary, False) 
52          self.assertEqual(hok.cache_grid, True) 
53          self.assertEqual(VtkSave.preloop, Hook.preloop) 
54          self.assertEqual(VtkSave.premarch, Hook.premarch) 
55          self.assertEqual(VtkSave.postmarch, Hook.postmarch) 
56          self.assertEqual(VtkSave.postloop, Hook.postloop) 
57   
58 -    def test_splitsave(self): 
59          from ..hook import Hook, SplitSave 
60          self.assertNotEqual(SplitSave.preloop, Hook.preloop) 
61          self.assertEqual(SplitSave.premarch, Hook.premarch) 
62          self.assertEqual(SplitSave.postmarch, Hook.postmarch) 
63          self.assertEqual(SplitSave.postloop, Hook.postloop) 
64   
65 -class BlockHookTest(TestCase): 
66 -    def setUp(self): 
67          self._msg = '' 
68   
69 -    def info(self, msg): 
70          self._msg += msg 
71   
72 -    def assertInfo(self, msg): 
73          self.assertEqual(self._msg, msg) 
74   
75 -class TestMarchSave(BlockHookTest): 
76 -    def test_methods(self): 
77          from ..hook import Hook, MarchSave 
78          self.assertTrue(isinstance(MarchSave.data, property)) 
79          self.assertTrue(callable(MarchSave._write)) 
80          self.assertNotEqual(MarchSave.preloop, Hook.preloop) 
81          self.assertEqual(MarchSave.premarch, Hook.premarch) 
82          self.assertNotEqual(MarchSave.postmarch, Hook.postmarch) 
83          self.assertEqual(MarchSave.postloop, Hook.postloop) 
84