class test_positive(object): ''' Test for the positive function in run_trim. Because it's hard (impossible?) to exactly match two floats we look at the detla. ''' def __init__(self): #self.input_array = N.resize(range(-99,1),(10,10)) self.input_array = N.array([-1,0]) self.PNGCreator = PNGCreator(self.input_array) def positive_test(self): self.PNGCreator.positive() result = self.PNGCreator.data expected = N.array([0,1]) + 0.0001 difference = round(abs(N.sum(result - expected))) assert difference == 0, 'positive is not working.'
def __init__(self): #self.input_array = N.resize(range(-99,1),(10,10)) self.input_array = N.array([-1,0]) self.PNGCreator = PNGCreator(self.input_array)