Fixed test failures in add_legend

This PR fixes errors in the test suite when asserting warnings for an empty legend. matplotlib 2.1 doesn't seem to warn any more...