Make the unit test on a custom file a cli arg.

Instead of modifying the source code, testing on a custom file can be
done by passing a filename as an argument.