Skip to content

Commit

Permalink
TestPoolLayer
Browse files Browse the repository at this point in the history
  • Loading branch information
pchavanne committed Oct 31, 2016
1 parent 803e19a commit 9138b5b
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tests/test_layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -310,11 +310,7 @@ def input_layer(self, input_data):
def layer(self, pool_layer, input_layer):
return pool_layer(input_layer, poolsize=(2, 2))

def test_output_shape(self, layer):
assert layer.output_shape() == (layer.input_shape[0],
layer.input_shape[1],
layer.input_shape[2] / layer.poolsize[0],
layer.input_shape[3] / layer.poolsize[1])


class TestConvLayer:
@pytest.fixture
Expand Down

0 comments on commit 9138b5b

Please sign in to comment.