Add a test to verify demo code runs as expected