Write tests for memory usage test

There are different types of outputs for memory tests, and we need unit tests to make sure we don't break one case while fixing the other.

Please write tests for all cases you know (issues reported, etc.)