Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Restructured queue.py and stack.py #152

Merged
merged 15 commits into from
Mar 16, 2020
Merged
Show file tree
Hide file tree
Changes from 9 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 18 additions & 29 deletions pydatastructs/miscellaneous_data_structures/queue.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from pydatastructs.linear_data_structures import DynamicOneDimensionalArray, SinglyLinkedList
from pydatastructs.utils.misc_util import NoneType, LinkedListNode
from pydatastructs.utils.misc_util import NoneType, LinkedListNode, _check_type
from copy import deepcopy as dc

__all__ = [
Expand Down Expand Up @@ -48,8 +48,7 @@ def __new__(cls, implementation='array', **kwargs):
kwargs.get('dtype', int))
elif implementation == 'linkedlist':
return LinkedListQueue(
kwargs.get('items', None),
kwargs.get('dtype', NoneType)
kwargs.get('items', None)
)
raise NotImplementedError(
"%s hasn't been implemented yet."%(implementation))
Expand Down Expand Up @@ -94,7 +93,7 @@ def append(self, x):

def popleft(self):
if self.is_empty:
raise ValueError("Queue is empty.")
raise IndexError("Queue is empty.")
return_value = dc(self.items[self.front])
front_temp = self.front
if self.front == self.rear:
Expand Down Expand Up @@ -128,53 +127,43 @@ def __str__(self):

class LinkedListQueue(Queue):

__slots__ = ['front', 'rear', 'size', '_dtype']
HarsheetKakar marked this conversation as resolved.
Show resolved Hide resolved

def __new__(cls, items=None, dtype=NoneType):
def __new__(cls, items=None):
obj = object.__new__(cls)
obj.queue = SinglyLinkedList()
obj._dtype = dtype
if items is None:
pass
elif type(items) in (list, tuple):
if len(items) != 0 and dtype is NoneType:
obj._dtype = type(items[0])
for x in items:
if type(x) == obj._dtype:
obj.queue.append(x)
else:
raise TypeError("Expected %s but got %s"%(obj._dtype, type(x)))
obj.append(x)
else:
raise TypeError("Expected type: list/tuple")
obj.front = obj.queue.head
obj.rear = obj.queue.tail
obj.size = obj.queue.size
return obj

def append(self, x):
if self._dtype is NoneType:
self._dtype = type(x)
elif type(x) is not self._dtype:
raise TypeError("Expected %s but got %s"%(self._dtype, type(x)))
self.size += 1
self.queue.append(x)
if self.front is None:
self.front = self.queue.head
self.rear = self.queue.tail

def popleft(self):
if self.is_empty:
raise ValueError("Queue is empty.")
self.size -= 1
raise IndexError("Queue is empty.")
return_value = self.queue.pop_left()
self.front = self.queue.head
self.rear = self.queue.tail
return return_value

@property
def is_empty(self):
return self.size == 0

@property
def front(self):
return self.queue.head

@property
def rear(self):
return self.queue.tail

@property
def size(self):
return self.queue.size

def __len__(self):
return self.size

Expand Down
11 changes: 8 additions & 3 deletions pydatastructs/miscellaneous_data_structures/stack.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,11 +64,13 @@ def pop(self, *args, **kwargs):

@property
def is_empty(self):
return None
raise NotImplementedError(
"This is an abstract method.")

@property
def peek(self):
return None
raise NotImplementedError(
"This is an abstract method.")

class ArrayStack(Stack):

Expand All @@ -90,7 +92,7 @@ def push(self, x):

def pop(self):
if self.is_empty:
raise ValueError("Stack is empty")
raise IndexError("Stack is empty")

top_element = dc(self.items[self.items._last_pos_filled])
self.items.delete(self.items._last_pos_filled)
Expand All @@ -104,6 +106,9 @@ def is_empty(self):
def peek(self):
return self.items[self.items._last_pos_filled]

def __len__(self):
return self.items.size
HarsheetKakar marked this conversation as resolved.
Show resolved Hide resolved

def __str__(self):
"""
Used for printing.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,10 @@ def test_Queue():
assert len(q1) == 0

q1 = Queue()
raises(ValueError, lambda: q1.popleft())
raises(IndexError, lambda: q1.popleft())

q1 = Queue(implementation='linkedlist')
q1.append(1)
assert raises(TypeError, lambda: q1.append('a'))
assert raises(TypeError, lambda: Queue(implementation='linkedlist', items=[0], dtype=str))
assert raises(TypeError, lambda: Queue(implementation='linkedlist', items={0, 1}))
q1 = Queue(implementation='linkedlist', items = [0, 1])
q1.append(2)
Expand All @@ -35,4 +33,7 @@ def test_Queue():
assert q1.popleft().data == 2
assert q1.popleft().data == 3
assert len(q1) == 0
raises(ValueError, lambda: q1.popleft())
raises(IndexError, lambda: q1.popleft())

q1 = Queue(implementation='linkedlist',items=['a',None,type,{}])
czgdp1807 marked this conversation as resolved.
Show resolved Hide resolved
assert len(q1) == 4
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,8 @@ def test_Stack():
assert s.pop() == 2
assert s.pop() == 1
assert s.is_empty is True
assert raises(ValueError, lambda : s.pop())
assert raises(IndexError, lambda : s.pop())
_s = Stack(items=[1, 2, 3])
assert str(_s) == '[1, 2, 3]'
assert len(_s) == 3
assert raises(NotImplementedError, lambda: Stack(implementation=''))