Added extra unit tests to chain_test.py