Add actual tests to test_pipeline.py