single_baseline_test.py updated: works at reasonable speed with beam