test: add a test for count output