def test_conv_get_config_has_all_attributes(): block = blocks.ConvBlock() config = block.get_config() assert utils.get_func_args( blocks.ConvBlock.__init__).issubset(config.keys())
def test_conv_build_return_tensor(): block = blocks.ConvBlock() outputs = block.build(kerastuner.HyperParameters(), tf.keras.Input(shape=(32, 32, 3), dtype=tf.float32)) assert len(nest.flatten(outputs)) == 1 assert isinstance(nest.flatten(outputs)[0], tf.Tensor)
def test_conv_build_with_dropout_return_tensor(): block = blocks.ConvBlock(dropout=0.5) outputs = block.build( keras_tuner.HyperParameters(), tf.keras.Input(shape=(32, 32, 3), dtype=tf.float32), ) assert len(nest.flatten(outputs)) == 1
def test_conv_with_small_image_size_return_tensor(): block = blocks.ConvBlock() outputs = block.build( keras_tuner.HyperParameters(), tf.keras.Input(shape=(10, 10, 3), dtype=tf.float32), ) assert len(nest.flatten(outputs)) == 1
def test_conv_deserialize_to_conv(): serialized_block = blocks.serialize(blocks.ConvBlock()) block = blocks.deserialize(serialized_block) assert isinstance(block, blocks.ConvBlock)