def test_bert_get_config_has_all_attributes():
block = blocks.BertBlock()
config = block.get_config()
assert utils.get_func_args(blocks.BertBlock.__init__).issubset(
config.keys())
def test_bert_build_return_tensor():
block = blocks.BertBlock()
outputs = block.build(keras_tuner.HyperParameters(),
tf.keras.Input(shape=(1, ), dtype=tf.string))
assert len(nest.flatten(outputs)) == 1
def test_bert_deserialize_to_transformer():
serialized_block = blocks.serialize(blocks.BertBlock())
block = blocks.deserialize(serialized_block)
assert isinstance(block, blocks.BertBlock)