Fix error which removes all elements from queue

1 files changed, 14 insertions, 0 deletions

diff --git a/queue-test.py b/queue-test.py
index c68ec51..131c2c5 100644
--- a/queue-test.py
+++ b/queue-test.py
@@ -12,3 +12,17 @@ def test_queue_average():
 def test_queue_average_null():
     a = circular_queue(50)
     assert a.average() == 0
+
+def test_queue_average_size():
+    a = circular_queue(50)
+    print(len(a.queue))
+    a.average()
+    print(len(a.queue))
+    assert len(a.queue) == 50
+
+def test_queue_circle():
+    a = circular_queue(20)
+    for i in range(0,2000): 
+        a.add(1)
+    assert a.average() == 1
+