Add some more unit tests for traces