Added additional tests to test_cosmicray.py