| 72 | |
| 73 | |
| 74 | def test_collecting_nodes(): |
| 75 | x = [1, 2, 3, 4, 5, 6, 7, 8] |
| 76 | wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric') |
| 77 | |
| 78 | # All nodes in natural order |
| 79 | assert_([node.path for node in wp.get_level(3, 'natural')] == |
| 80 | ['aaa', 'aad', 'ada', 'add', 'daa', 'dad', 'dda', 'ddd']) |
| 81 | |
| 82 | # and in frequency order. |
| 83 | assert_([node.path for node in wp.get_level(3, 'freq')] == |
| 84 | ['aaa', 'aad', 'add', 'ada', 'dda', 'ddd', 'dad', 'daa']) |
| 85 | |
| 86 | assert_raises(ValueError, wp.get_level, 3, 'invalid_order') |
| 87 | |
| 88 | |
| 89 | def test_reconstructing_data(): |