Improved legend test coverage

I was looking over the test coverage report from #273, and noticed we were missing a number of legend tests.