Bring back unit tests

[samihult]

• Make unit tests run prior to publishing
• Remove outdated unit tests
• Move unit test to packages where they belong, if needed
• Make sure all unit tests pass