Move some demos to unit tests