SPIKE: investigate ways to slim down Python file used for testing

[cquil11]

cc @kimbochen

The refactor in #145 introduced a singular Python script with lots of logic. We can try to simplify it as a p2 task.