Skip to content

Commit 3590cb1

Browse files
committed
Add unittests
1 parent 057d0cb commit 3590cb1

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

python/paddle/v2/tests/test_layer.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import paddle.v2.data_type as data_type
1919
import paddle.v2.layer as layer
2020
import paddle.v2.pooling as pooling
21+
import paddle.v2.networks as networks
2122

2223
pixel = layer.data(name='pixel', type=data_type.dense_vector(128))
2324
label = layer.data(name='label', type=data_type.integer_value(10))
@@ -251,5 +252,13 @@ def test_operator(self):
251252
print layer.parse_network(conv1)
252253

253254

255+
class NetworkTests(unittest.TestCase):
256+
def test_vgg(self):
257+
img = layer.data(name='pixel', type=data_type.dense_vector(784))
258+
vgg_out = networks.small_vgg(
259+
input_image=img, num_channels=1, num_classes=2)
260+
print layer.parse_network(vgg_out)
261+
262+
254263
if __name__ == '__main__':
255264
unittest.main()

0 commit comments

Comments
 (0)