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_divideArray(self): actual = divideArray() self.assertEqual(1.6, actual[1, 3]) self.assertEqual(1.1, actual[2, 1]) self.assertEqual(1.3, actual[2, 3]) self.assertEqual(2., actual[0, 2])