Fix error which removes all elements from queue

1 files changed, 8 insertions, 6 deletions

diff --git a/circular_queue.py b/circular_queue.py
index 44954bd..6b36748 100644
--- a/circular_queue.py
+++ b/circular_queue.py
@@ -11,11 +11,13 @@ class circular_queue:
         self.pointer += 1
         if self.pointer >= self.size:
             self.pointer = 0
-
     def average(self):
-        queue = self.queue
-        while None in queue:
-            queue.remove(None)
-        if len(queue) == 0:
+        value = 0
+        length = 0
+        for i in self.queue:
+            if i != None:
+                value += i
+                length += 1
+        if length == 0:
             return 0
-        return sum(queue) / len(queue)
+        return value / length