IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Browse Source

Add test_[one_hot_]input_layer_parameter_count_01().

master
Stanislaw Adaszewski 3 years ago
parent
commit
1b72e41ea9
1 changed files with 13 additions and 0 deletions
  1. +13
    -0
      tests/icosagon/test_input.py

+ 13
- 0
tests/icosagon/test_input.py View File

@@ -107,3 +107,16 @@ def test_one_hot_input_layer_04():
layer = OneHotInputLayer(d)
s = repr(layer)
assert s.startswith('Icosagon one-hot input layer')
def test_one_hot_input_layer_parameter_count_01():
d = _some_data()
layer = OneHotInputLayer(d)
assert len(list(layer.parameters())) == 2
def test_input_layer_parameter_count_01():
d = _some_data()
for output_dim in [32, 64, 128]:
layer = InputLayer(d, output_dim)
assert len(list(layer.parameters())) == 2

Loading…
Cancel
Save