Write tests for main.py and make_dataset.py.

[ambuc]

An integration test which runs continuously on Github Actions would build confidence that this code doesn't break over time.