def test_devectorize_axes(): np.random.seed(0) x, y = np.random.random((2, 1000)) # save vectorized version fig = plt.figure() ax = fig.add_subplot(111) ax.scatter(x, y) output = BytesIO() fig.savefig(output) output.seek(0) im1 = image.imread(output) plt.close() # save devectorized version fig = plt.figure() ax = fig.add_subplot(111) ax.scatter(x, y) devectorize_axes(ax, dpi=200) output = BytesIO() fig.savefig(output) output.seek(0) im2 = image.imread(output) plt.close() assert_(im1.shape == im2.shape) assert_((im1 != im2).sum() < 0.1 * im1.size)
def test_devectorize_axes(): np.random.seed(0) x, y = np.random.random((2, 1000)) # save vectorized version fig = plt.figure() ax = fig.add_subplot(111) ax.scatter(x, y) sio = StringIO() fig.savefig(sio) sio.reset() im1 = image.imread(sio) plt.close() # save devectorized version fig = plt.figure() ax = fig.add_subplot(111) ax.scatter(x, y) devectorize_axes(ax, dpi=200) sio = StringIO() fig.savefig(sio) sio.reset() im2 = image.imread(sio) plt.close() assert_(im1.shape == im2.shape) assert_((im1 != im2).sum() < 0.1 * im1.size)
mag_z += -0.005 + 0.01 * np.random.random(size=mag_z.shape) # compute RGB color based on magnitudes color = compute_color(mag_a, mag_i, mag_z) #------------------------------------------------------------ # set up the plot fig = plt.figure(figsize=(10.5, 5), facecolor='k') fig.subplots_adjust(left=0.1, right=0.95, wspace=0.2, bottom=0.1, top=0.9) # plot the color-magnitude plot ax = black_bg_subplot(121) ax.scatter(mag_a, mag_i - mag_z, c=color, s=1, lw=0) devectorize_axes(ax, dpi=400) ax.plot([0, 0], [-0.8, 0.6], '--w', lw=2) ax.plot([0, 0.4], [-0.15, -0.15], '--w', lw=2) ax.set_xlim(-0.3, 0.4) ax.set_ylim(-0.8, 0.6) ax.set_xlabel('a*', color='w') ax.set_ylabel('i-z', color='w') # plot the orbital parameters plot ax = black_bg_subplot(122) ax.scatter(a, sini, c=color, s=1, lw=0) devectorize_axes(ax, dpi=400)