FIX: Adding viz tests
ENH: Adding test
Merge pull request #397 from Eric89GXL/viz-tests

FIX: Adding viz tests