Add tests for setYScales functionality