import io                                                             
import time                                                           
import numpy as np                                                    
from mpl_toolkits.mplot3d import Axes3D                               
import matplotlib.pyplot as plt                                             
from contextlib import contextmanager                                 
from collections import defaultdict                                   
                                                                      
results = defaultdict(lambda: np.zeros((2,)))                         
@contextmanager                                                       
def measure(name):                                                    
    s = time.perf_counter()                                           
    yield                                                             
    d = time.perf_counter() - s                                       
    results[name] += [1, d]                                           
                                                                      
def summarize():                                                      
    for name, (count, total) in sorted(results.items()):              
        print(name, total / count)                                    
                                                                      
x = y = np.linspace(-1, 1, 400)                                       
x, y = np.meshgrid(x, y)                                              
z = x ** 2 + y ** 2                                                   
                                                                      
                                                                      
for i in range(4):                                                    
    fig = plt.figure()                                                
    ax = fig.gca(projection='3d')                                     
    with measure('plot'):                                             
        ax.plot_surface(x, y, z, rstride=1, cstride=1)
    with measure('draw'):                                             
        buf = io.BytesIO()                                            
        fig.savefig(buf, format='raw')                                
    plt.close()                                                       
                                                                      
summarize()