def testDenseBridge(self): encoder_state = _build_state(3, 20, 6) decoder_state = _build_state(4, 30, 6) state = bridge.DenseBridge()(encoder_state, decoder_state) bridge.assert_state_is_compatible(decoder_state, state)