Improve unit testing of do()

[mscroggs]

Would be better to do some more specific unit tests rather than just testing expection

See suggestions in #42 (comment)

• Experiment with making a new graph with do() and calling numpyro's do
• Test that numpyro's do and do do the same thing #91