Testing matplotlib create graph class