Skip to content

Commit

Permalink
first yadll commit
Browse files Browse the repository at this point in the history
  • Loading branch information
pchavanne committed Jul 27, 2016
1 parent 225a296 commit 6fe0cc9
Showing 1 changed file with 16 additions and 1 deletion.
17 changes: 16 additions & 1 deletion tests/test_layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -289,12 +289,27 @@ def test_get_output(self, input_layer, layer, layer_c0, layer_c1):
assert np.all(layer_c1.get_output().eval() == 0)


class TestDropPoolLayer:
class TestPoolLayer:
@pytest.fixture
def pool_layer(self):
from yadll.layers import PoolLayer
return PoolLayer

@pytest.fixture
def input_data(self):
from yadll.utils import shared_variable
return shared_variable(np.random.random((10, 20)))

@pytest.fixture
def input_layer(self, input_data):
from yadll.layers import InputLayer
shape = (10, 20)
return InputLayer(shape, input_var=input_data)

@pytest.fixture
def layer(self, pool_layer, input_layer):
return pool_layer(input_layer, poolsize=(2, 2))


class TestConvLayer:
@pytest.fixture
Expand Down

0 comments on commit 6fe0cc9

Please sign in to comment.