I think it could be better as it is a test to illustrate the behaviour. it is subjective though but logically it feels more like a test than part of the app