def test_remove(self): ul = UnorderedList() ul.add(Node('first')) ul.add(Node('second')) ul.add(Node('third')) self.assertEqual(ul.head.next.data, 'second') ul.remove('second') self.assertEqual(ul.head.next.data, 'first')