split test_flow_logic.py into 4 files