mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-03 16:08:02 +00:00
36 lines
833 B
Python
36 lines
833 B
Python
from nose.tools import assert_equal
|
|
|
|
|
|
class TestSetOfStacks(object):
|
|
|
|
def test_set_of_stacks(self):
|
|
print('Test: Push on an empty stack')
|
|
stacks = SetOfStacks(capacity=2)
|
|
stacks.push(3)
|
|
|
|
print('Test: Push on a non-empty stack')
|
|
stacks.push(5)
|
|
|
|
print('Test: Push on a capacity stack to create a new one')
|
|
stacks.push('a')
|
|
|
|
print('Test: Pop on a stack to destroy it')
|
|
assert_equal(stacks.pop(), 'a')
|
|
|
|
print('Test: Pop general case')
|
|
assert_equal(stacks.pop(), 5)
|
|
assert_equal(stacks.pop(), 3)
|
|
|
|
print('Test: Pop on no elements')
|
|
assert_equal(stacks.pop(), None)
|
|
|
|
print('Success: test_set_of_stacks')
|
|
|
|
|
|
def main():
|
|
test = TestSetOfStacks()
|
|
test.test_set_of_stacks()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |