Add test for origin-based prediction