def test_Q2(self):
a = np.arange(25).reshape(5, 5)
b = np.array([1., 5, 10, 15, 20])
c = md.matrix_divide(a, b)
d = np.array([[0., 1., 2., 3., 4.], [1., 1.2, 1.4, 1.6, 1.8],
[1., 1.1, 1.2, 1.3, 1.4], [1., 1.067, 1.133, 1.2, 1.267],
[1., 1.05, 1.1, 1.15, 1.2]])
self.assertTrue(np.array_equal(c, d))
def test_Q2(self):
a = np.arange(25).reshape(5, 5)
b = np.array([1., 5, 10, 15, 20])
c = md.matrix_divide(a,b)
d = np.array([[ 0., 1., 2., 3., 4.],
[1., 1.2, 1.4, 1.6, 1.8],
[1., 1.1, 1.2, 1.3, 1.4],
[ 1., 1.067, 1.133, 1.2, 1.267],
[ 1., 1.05, 1.1, 1.15, 1.2]])
self.assertTrue(np.array_equal(c,d))
print "\n The answer of Question 1, (b) is"
print m2day.make_2D_array(5, 3, 1, 15)[:, 1:2]
#c
print "\n The answer of Question 1, (c) is"
print m2day.make_2D_array(5, 3, 1, 15)[1:4, :3]
#d
print "\n The answer of Question 1, (d) is"
print m2day.make_2D_array(5, 3, 1, 15).T.reshape(1,15)[0][2:11]
# Question 2
a = np.arange(25).reshape(5, 5)
b = np.array([1., 5, 10, 15, 20])
print "\n The answer of Question 2 is"
print md.matrix_divide(a,b)
# Question 3
print "\n The answer of Question 3 is"
print cl5.close_to_5()
# Question 4
print "\n Generating plot.........."
mask = mand.mandelbort_set(-2, 1, -1.5, 1.5)
print "\n The plot of question 4 is saved as 'mandelbrot.png' in the current dictory."
plt.imshow(mask.T, extent=[-2, 1, -1.5, 1.5])
plt.gray()
plt.savefig('mandelbrot.png')
except KeyboardInterrupt, ValueError:
print "\n Interrupted!"
