mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-06 01:18:02 +00:00
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
from nose.tools import assert_equal
|
|
|
|
|
|
class TestLinkedList(object):
|
|
|
|
def test_insert_to_front(self):
|
|
print('Test: insert_to_front on an empty list')
|
|
linked_list = LinkedList(None)
|
|
linked_list.insert_to_front(10)
|
|
assert_equal(linked_list.get_all_data(), [10])
|
|
|
|
print('Test: insert_to_front on a None')
|
|
linked_list.insert_to_front(None)
|
|
assert_equal(linked_list.get_all_data(), [10])
|
|
|
|
print('Test: insert_to_front general case')
|
|
linked_list.insert_to_front('a')
|
|
linked_list.insert_to_front('bc')
|
|
assert_equal(linked_list.get_all_data(), ['bc', 'a', 10])
|
|
|
|
print('Success: test_insert_to_front\n')
|
|
|
|
def test_append(self):
|
|
print('Test: append on an empty list')
|
|
linked_list = LinkedList(None)
|
|
linked_list.append(10)
|
|
assert_equal(linked_list.get_all_data(), [10])
|
|
|
|
print('Test: append a None')
|
|
linked_list.append(None)
|
|
assert_equal(linked_list.get_all_data(), [10])
|
|
|
|
print('Test: append general case')
|
|
linked_list.append('a')
|
|
linked_list.append('bc')
|
|
assert_equal(linked_list.get_all_data(), [10, 'a', 'bc'])
|
|
|
|
print('Success: test_append\n')
|
|
|
|
def test_find(self):
|
|
print('Test: find on an empty list')
|
|
linked_list = LinkedList(None)
|
|
node = linked_list.find('a')
|
|
assert_equal(node, None)
|
|
|
|
print('Test: find a None')
|
|
head = Node(10)
|
|
linked_list = LinkedList(head)
|
|
node = linked_list.find(None)
|
|
assert_equal(node, None)
|
|
|
|
print('Test: find general case with matches')
|
|
head = Node(10)
|
|
linked_list = LinkedList(head)
|
|
linked_list.insert_to_front('a')
|
|
linked_list.insert_to_front('bc')
|
|
node = linked_list.find('a')
|
|
assert_equal(str(node), 'a')
|
|
|
|
print('Test: find general case with no matches')
|
|
node = linked_list.find('aaa')
|
|
assert_equal(node, None)
|
|
|
|
print('Success: test_find\n')
|
|
|
|
def test_delete(self):
|
|
print('Test: delete on an empty list')
|
|
linked_list = LinkedList(None)
|
|
linked_list.delete('a')
|
|
assert_equal(linked_list.get_all_data(), [])
|
|
|
|
print('Test: delete a None')
|
|
head = Node(10)
|
|
linked_list = LinkedList(head)
|
|
linked_list.delete(None)
|
|
assert_equal(linked_list.get_all_data(), [10])
|
|
|
|
print('Test: delete general case with matches')
|
|
head = Node(10)
|
|
linked_list = LinkedList(head)
|
|
linked_list.insert_to_front('a')
|
|
linked_list.insert_to_front('bc')
|
|
linked_list.delete('a')
|
|
assert_equal(linked_list.get_all_data(), ['bc', 10])
|
|
|
|
print('Test: delete general case with no matches')
|
|
linked_list.delete('aa')
|
|
assert_equal(linked_list.get_all_data(), ['bc', 10])
|
|
|
|
print('Success: test_delete\n')
|
|
|
|
def test_len(self):
|
|
print('Test: len on an empty list')
|
|
linked_list = LinkedList(None)
|
|
assert_equal(len(linked_list), 0)
|
|
|
|
print('Test: len general case')
|
|
head = Node(10)
|
|
linked_list = LinkedList(head)
|
|
linked_list.insert_to_front('a')
|
|
linked_list.insert_to_front('bc')
|
|
assert_equal(len(linked_list), 3)
|
|
|
|
print('Success: test_len\n')
|
|
|
|
|
|
def main():
|
|
test = TestLinkedList()
|
|
test.test_insert_to_front()
|
|
test.test_append()
|
|
test.test_find()
|
|
test.test_delete()
|
|
test.test_len()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |