Add unit tests to Testability Demo