(self)
| 89 | ) |
| 90 | |
| 91 | def test_beam_queue(self): |
| 92 | beam_queue = alphageometry.BeamQueue(max_size=2) |
| 93 | |
| 94 | beam_queue.add('a', 1) |
| 95 | beam_queue.add('b', 2) |
| 96 | beam_queue.add('c', 3) |
| 97 | |
| 98 | beam_queue = list(beam_queue) |
| 99 | self.assertEqual(beam_queue, [(3, 'c'), (2, 'b')]) |
| 100 | |
| 101 | |
| 102 | if __name__ == '__main__': |