Skip to content

Commit

Permalink
Reduce length of conv UT
Browse files Browse the repository at this point in the history
  • Loading branch information
ngoguey committed Aug 12, 2020
1 parent 382a1c5 commit 0fac7bb
Showing 1 changed file with 9 additions and 19 deletions.
28 changes: 9 additions & 19 deletions buzzard/test/test_footprint_convs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand All @@ -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),
])

# *********************************************************************** **
Expand Down

0 comments on commit 0fac7bb

Please sign in to comment.