Add notebook smoke tests

[ed1d1a8d]

Can just copy the ones from here for now: https://github.com/HumanCompatibleAI/go_attack/blob/main/tests/test_examples.py