def test_size(self):
my_stack = Stack()
my_stack.push(1)
my_stack.push(2)
my_stack.push(3)
data = my_stack.size()
res = 3
self.assertEqual( data, res)
def test_random(self):
for i in range(10000):
s = Stack()
res = []
comand = random.randint(0, 1) # 0 - pop, 1 - push
if comand:
t = random.randint(0,9)
s.push(t)
res.append(t)
else:
s.pop()
try:
res.pop()
except:
pass
data = s.stack + [s.size()]
res = res[::-1] + [len(res)]
self.assertEqual( data, res)
def testSize(self):
s = Stack()
s.add("1")
self.assertTrue(s.size() == 1)
s.remove()
self.assertFalse(s.size() == 1)
def testRemove(self):
s = Stack()
s.add("1")
s.add("2")
s.remove()
self.assertTrue(s.size() == 1)
def testAdd(self):
s = Stack()
s.add("1")
s.add("2")
self.assertTrue(s.size() == 2)
