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)
