Adding matplotlib tests
organizing tests