Guitar-String-Algorithm/ringbuffer_tester.py at main · newonathe/Guitar-String-Algorithm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#!/usr/bin/env python3

from ringbuffer import *
import unittest

class RingBufferTester(unittest.TestCase):
    def rbtest__init(self):
        buf = RingBuffer(69)
        self.assertEqual(buf.MAX_CAP, 69)
        self.assertEqual(buf.size(), 0)

    def rbtest_buffer_isa_list(self):
        buf = RingBuffer(69)
        self.assertTrue(isinstance(buf.buffer, list))

    def rbtest_enqueue_1(self):
        buf = RingBuffer(20)
        buf.enqueue(69)
        x = buf.peek()
        self.assertEqual(x, 69)

    def rbtest_dequeue_1(self):
        buf = RingBuffer(20)
        buf.enqueue(5)
        x = buf.dequeue()
        self.assertEqual(x, 5)

    def rbtest_ordering(self):
        buf = RingBuffer(20)
        for i in range(10):
            buf.enqueue(i)
        for i in range(10):
            x = buf.dequeue()
            self.assertEqual(x, i)

    def rbtest_empty_check(self):
        buf = RingBuffer(10)
        self.assertTrue(buf.is_empty())

    def rbtest_full_check(self):
        buf = RingBuffer(5)
        for _ in range(5):
            buf.enqueue(0)
        self.assertTrue(buf.is_full())

    def rbtest_enqueue_full_error(self):
        buf = RingBuffer(5)
        for _ in range(5):
            buf.enqueue(0)
        with self.assertRaises(RingBufferFull):
            buf.enqueue(0)

    def rbtest_peek_empty_error(self):
        buf = RingBuffer(10)
        with self.assertRaises(RingBufferEmpty):
            buf.peek()

    def rbtest_dequeue_empty_error(self):
        buf = RingBuffer(10)
        with self.assertRaises(RingBufferEmpty):
            buf.dequeue()


if __name__ == '__main__':
    unittest.defaultTestLoader.testMethodPrefix = 'rbtest'
    unittest.main()