Remove bivariate_norm from mpl3d test code

dstansby · dstansby · commit 1b348bed069e · 2018-01-29T21:14:31.000Z
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2952,13 +2952,12 @@ def get_test_data(delta=0.05):
     '''
     Return a tuple X, Y, Z with a test data set.
     '''
-
-    from matplotlib.mlab import  bivariate_normal
     x = y = np.arange(-3.0, 3.0, delta)
     X, Y = np.meshgrid(x, y)
 
-    Z1 = bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
-    Z2 = bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
+    Z1 = np.exp(-(X**2 + Y**2) / 2) / (2 * np.pi)
+    Z2 = (np.exp(-(((X - 1) / 1.5)**2 + ((Y - 1) / 0.5)**2) / 2) /
+          (2 * np.pi * 0.5 * 1.5))
     Z = Z2 - Z1
 
     X = X * 10
@@ -2967,7 +2966,6 @@ def get_test_data(delta=0.05):
     return X, Y, Z
 
 
-
 ########################################################
 # Register Axes3D as a 'projection' object available
 # for use just like any other axes
