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()
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