For future reference, the built-in unittest library comes packed with assertion methods to make your life easier. Also, when they fail they provide better messages for what part of the test failed.

http://docs.python.org/2/library/unittest.html#unittest.TestCase.assertEqual

for instance this line could be
```python
self.assertIn('def add_arrays(a,b)', src)
```