From 0fac7bbcc23b7ad79893b1a35650a12d7299e2c4 Mon Sep 17 00:00:00 2001 From: ngoguey Date: Wed, 12 Aug 2020 11:07:01 +0200 Subject: [PATCH] Reduce length of conv UT --- buzzard/test/test_footprint_convs.py | 28 +++++++++------------------- 1 file changed, 9 insertions(+), 19 deletions(-) diff --git a/buzzard/test/test_footprint_convs.py b/buzzard/test/test_footprint_convs.py index a638613..6eff728 100644 --- a/buzzard/test/test_footprint_convs.py +++ b/buzzard/test/test_footprint_convs.py @@ -35,16 +35,13 @@ def _pytest_generate_tests(metafunc): ] if 'output_padding' in metafunc.fixturenames: rsizes = [ - (4, 4), - (5, 5), + (4, 5), (6, 7), ] else: rsizes = [ - (15, 15), - (16, 16), - (17, 17), - (18, 19), + (15, 16), + (17, 18), ] src_fp = buzz.Footprint(rsize=(1, 1), gt=(100, 1, 0, 100, 0, 1)) confs = list(itertools.product(scales, rots, rsizes)) @@ -63,45 +60,38 @@ def _pytest_generate_tests(metafunc): # *********************************************************************** ** kernel_sizes = np.asarray([ - (1, 1), - (2, 2), - (3, 3), - (4, 5), + (1, 2), + (3, 4), ]) # *********************************************************************** ** strides = np.asarray([ - (1, 1), - (2, 2), + (1, 2), (3, 4), ]) # *********************************************************************** ** if 'output_padding' in metafunc.fixturenames: paddings = np.asarray([ - (0, 0), - (1, 1), (0, 1), + (2, 3), ]) else: paddings = np.asarray([ - (0, 0), - (1, 1), + (0, 1), (2, 3), ]) # *********************************************************************** ** dilations = np.asarray([ (1, 1), - (2, 2), (2, 3), ]) # *********************************************************************** ** output_paddings = np.asarray([ - (0, 0), - (1, 1), (0, 1), + (2, 3), ]) # *********************************************************************** **