def test_reasonable_values(self): """Test using some reasonable values to reduce lines found""" # Given window = get_transform_window(transforms.HoughLinesP, IMG_PATH) gauss, canny, hough = window.transforms[1:4] # Set reasonable values for gauss.k_size_x = 9 gauss.k_size_y = 9 gauss.sigma_x = 2 canny.threshold1 = 185 canny.threshold2 = 120 canny.aperture_size = 3 canny.use_l2_gradient = True hough.rho = 159 hough.theta = 1.38 hough.threshold = 36 hough.min_length = 160 hough.max_gap = 59 # When window.draw(None, None) # Then _check_error(window)
def test_defaults(self): """Test using the defaults for all transforms""" # Given window = get_transform_window(transforms.HoughCircles, IMG_PATH) # When window.draw(None, None) # Then _check_error(window)
def test_merge(self): """No parameters here""" # Given window = get_transform_window(transforms.Split, IMG_PATH) # When window.draw(None, None) # Then _check_error(window)
def test_various_values(self, value): # Given window = get_transform_window(transforms.MedianBlur, IMG_PATH) tf = window.transforms[1] tf.k_size = value # When window.draw(None, None) # Then assert tf.error is None
def test_kernel_values(self, value): # Given window = get_transform_window(transforms.Filter2D, IMG_PATH) tf = window.transforms[1] tf.kernel = value # When window.draw(None, None) # Then _check_error(window)
def test_kernel_sizes(self, rows, cols): # Given window = get_transform_window(transforms.Filter2D, IMG_PATH) tf = window.transforms[1] tf.kernel = np.ones((rows, cols)) # When window.draw(None, None) # Then _check_error(window)
def _test_single_attr(transform, attr, value, tf_num=1): """Tests setting single attribute and running pipeline""" # Given window = get_transform_window(transform, IMG_PATH) tf = window.transforms[tf_num] setattr(tf, attr, value) # When window.draw(None, None) # Then _check_error(window)
def test_anchors(self, row, col): # Given window = get_transform_window(transforms.Filter2D, IMG_PATH) tf = window.transforms[1] tf.kernel = np.ones((3, 3)) tf._kernel.anchor = (row, col) # When window.draw(None, None) # Then _check_error(window)
def test_l2(self, alpha): # Given window = get_transform_window(transforms.Normalize, IMG_PATH) tf = window.transforms[1] tf.alpha = alpha tf.norm_type = cvc.NORMS['NORM_L2'] # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, n_images): """Test Other params""" # Given window = get_transform_window(transforms.PyrDown, IMG_PATH) tf = window.transforms[1] tf.n_images = n_images # When window.draw(None, None) # Then _check_error(window)
def test_border_types(self, border_type): """Test the different border options""" # Given window = get_transform_window(transforms.Sobel, IMG_PATH) tf = window.transforms[1] tf.border_type = cvc.BORDERS[border_type] # When window.draw(None, None) # Then _check_error(window)
def test_border_types(self, border_type): """Test the different border options""" # Given window = get_transform_window(transforms.Dilate, IMG_PATH) tf = window.transforms[1] tf.iterations = 5 tf.border_type = cvc.BORDERS[border_type] tf.border_val = (255, 255, 255) # When window.draw(None, None) # Then _check_error(window)
def test_method_options(self, method): """Test method options""" window = get_transform_window(transforms.FindContours, IMG_PATH) tf = window.transforms[1] tf.method = cvc.CHAIN_APPROX[method] draw_cont = window.transforms[2] draw_cont.enabled = False # When window.draw(None, None) # Then _check_error(window)
def test_mode_options(self, mode): """Test Mode options""" window = get_transform_window(transforms.FindContours, IMG_PATH) tf = window.transforms[1] tf.mode = cvc.RETR[mode] draw_cont = window.transforms[2] draw_cont.enabled = False # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, scale_x, scale_y): """Test Other params""" # Given window = get_transform_window(transforms.Resize, IMG_PATH) tf = window.transforms[1] tf.scale_x = scale_x tf.scale_y = scale_y # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, kernel, normalize): """Test other param combinations""" # Given window = get_transform_window(transforms.BoxFilter, IMG_PATH) tf = window.transforms[1] tf.kernel = kernel tf.normalize = normalize # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, r, theta): """Test other param combinations""" # Given window = get_transform_window(transforms.Remap, IMG_PATH) tf = window.transforms[1] tf.r = r tf.theta = theta # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, alpha, beta, gamma): """Test Other params""" # Given window = get_transform_window(transforms.AddWeighted, IMG_PATH) tf = window.transforms[1] tf.alpha = alpha tf.beta = beta tf.gamma = gamma # When window.draw(None, None) # Then _check_error(window)
def test_min_and_middle_val(self, value): # Given window = get_transform_window(transforms.CopyMakeBorder, IMG_PATH) tf = window.transforms[1] tf.top = value tf.right = value tf.bottom = value tf.left = value # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, ch1, ch2, ch3): """Test other param combinations""" # Given window = get_transform_window(transforms.InRangeRaw, IMG_PATH) tf = window.transforms[1] tf.ch1['top'], tf.ch1['bot'] = ch1 tf.ch2['top'], tf.ch2['bot'] = ch2 tf.ch3['top'], tf.ch3['bot'] = ch3 # When window.draw(None, None) # Then _check_error(window)
def test_canny(self, thresh1, thresh2, aperture, l2_grad): # Given window = get_transform_window(transforms.Canny, IMG_PATH) tf = window.transforms[1] tf.threshold1 = thresh1 tf.threshold2 = thresh2 tf.aperture_size = aperture tf.use_l2_gradient = l2_grad # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, kernel_x, kernel_y, delta): """Test other param combinations""" # Given window = get_transform_window(transforms.SepFilter2D, IMG_PATH) tf = window.transforms[1] tf.kernel_X = kernel_x tf.kernel_Y = kernel_y tf.delta = delta # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, kernel, anchor, iterations): """Test combinations of the other params""" # Given window = get_transform_window(transforms.Dilate, IMG_PATH) tf = window.transforms[1] tf.iterations = iterations tf.kernel = kernel tf._kernel.anchor = anchor # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, thresh): """Test Other params""" # Given window = get_transform_window(transforms.FindContours, IMG_PATH) tf = window.transforms[1] tf.threshold = thresh draw_cont = window.transforms[2] draw_cont.enabled = False # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, diameter, sigma_c, sigma_s): """Test combinations of the other params""" # Given window = get_transform_window(transforms.BilateralFilter, IMG_PATH) tf = window.transforms[1] tf.d = diameter tf.sigma_color = sigma_c tf.sigma_space = sigma_s # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, criteria, win_rows, win_cols, eps, max_iter): """Test Other params""" # Given window = get_transform_window(transforms.CornerSubPix, IMG_PATH) tf = window.transforms[2] tf.criteria = cvc.TERM[criteria] tf.window_size_rows = win_rows tf.window_size_cols = win_cols tf.epsilon = eps tf.max_iter = max_iter # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, criteria, k, epsilon, max_iter, attempts): """Test Other params""" # Given window = get_transform_window(transforms.Kmeans, IMG_PATH) tf = window.transforms[1] tf.criteria = cvc.TERM[criteria] tf.k = k tf.epsilon = epsilon tf.max_iter = max_iter tf.attempts = attempts # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, temp_win_size, search_win_size, h, h_color): """Test other param combinations""" # Given window = get_transform_window(transforms.FastNIMeansDenoisingColored, IMG_PATH) tf = window.transforms[1] tf.template_window_size = temp_win_size tf.search_window_size = search_win_size tf.h = h tf.h_color = h_color # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, k_size, sigma): """Test Other params""" # Given window = get_transform_window(transforms.GetGaussianKernel, IMG_PATH) # Drop the display transform - window isn't set on pipeline # pipeline usually does that setup and we're not using it window.transforms.pop() tf = window.transforms[1] tf.k_size = k_size tf.sigma = sigma # When window.draw(None, None) # Then _check_error(window)
def test_other_params(self, dx, dy, k_size, scale, delta): """Test other param combinations""" # Given window = get_transform_window(transforms.Sobel, IMG_PATH) tf = window.transforms[1] tf.dx = dx tf.dy = dy tf.k_size = k_size tf.scale = scale tf.delta = delta # When window.draw(None, None) # Then _check_error(window)