Exemple #1
0
def main():

	# create an ArrayGenerator object and call the functions 
	print "Question1 initial 2-D array:"
	question_1 = ArrayGenerator()
	print question_1.added_array

	print "Question1 part a:"
	print question_1.rowArray() # question1 part a

	print "Question1 part b:"
	print question_1.colArray() # question1 part b

	print "Question1 part c:"
	print question_1.recArray() # question1 part c

	print "Question1 part d:"
	print question_1.restrictedArray() # question1 part d

	# question 2
	print "Question2:"
	print divideArray()

	# question 3
	print "Question3:"
	print randomArray()

	# question 4 
	print "Question4: the plot will be saved"
	compute_mandelbrot()
Exemple #2
0
	def test_randomArray(self):
		actual = randomArray()

		test_originalArray = np.random.rand(10,3)
		test_changedArray = np.argsort(abs(test_originalArray - 0.5))
		test_fancyIndex = test_changedArray[:, 1]
		test_final_array = test_originalArray[np.arange(0,10), test_fancyIndex]
		for i in range(10):
			if abs(test_final_array[i] - 0.5) >= abs(actual[i] - 0.5):
				self.assertTrue