Use assertEqual in examples/test_examples.py