def test_change_zoom_level_smaller_manual_check(self):
add_stochastic_noise(1, self.dim, self.arr)
cy.clear_bounds(self.dim, self.arr)
self.arr[0, 0] = 2 # make the zoomer resize
dim, buf, change, self.buf_stat = change_zoom_level(
self.position, self.buf_len, self.buf_stat, self.change_roll,
self.dim, self.buf)
nu_arr = buf[self.position[0]]
self.assertEqual(nu_arr[1, 1], 2)
self.assertEqual(change[0, 1], 1)
nu_arr[1, 1] = 0
nu_arr_mid = np.copy(nu_arr)
nu_arr[0, 0] = 2 #zoom again
dim, buf, change, self.buf_stat = change_zoom_level(
self.position, self.buf_len, self.buf_stat, change, dim, buf)
nu_arr = buf[self.position[0]]
self.assertEqual(nu_arr[1, 1], 2) #second assert
self.assertEqual(change[0, 1], 1)
nu_arr[1, 1] = 0
dim, buf, change, self.buf_stat = change_zoom_level(
self.position, self.buf_len, self.buf_stat, change, dim, buf)
nu_arr = buf[self.position[0]]
self.assertEqual(change[0, 1], -1)
testing.assert_array_equal(nu_arr, nu_arr_mid)
def test_clear_bounds(self):
arr = tst_arr()
cy.clear_bounds(tst_dim(), arr)
arr2 = tst_arr()
arr2[0, :] = 0
arr2[-1, :] = 0
arr2[:, 0] = 0
arr2[:, -1] = 0
testing.assert_array_equal(arr2, arr)
def test_change_zoom_level_same(self):
add_stochastic_noise(1, self.dim, self.arr)
cy.clear_bounds(self.dim, self.arr)
self.assertEqual(len(self.buf_stat), 1)
dim, buf, change, self.buf_stat = change_zoom_level(
self.position, self.buf_len, self.buf_stat, self.change_roll,
self.dim, self.buf)
self.assertEqual(change[0, 1], 0)
self.assertEqual(dim[0], self.dim[0])
testing.assert_array_equal(self.arr, buf[self.position[0]])
self.assertEqual(len(self.buf_stat), 1)