def testCopyBridgeSizeMismatch(self):
encoder_state = _build_state(3, 20, 6)
decoder_state = _build_state(3, 30, 6)
with self.assertRaises(ValueError):
_ = bridge.CopyBridge()(encoder_state, decoder_state)
def testCopyBridge(self):
encoder_state = _build_state(3, 20, 6)
decoder_state = _build_state(3, 20, 6)
state = bridge.CopyBridge()(encoder_state, decoder_state)
self.assertTupleEqual(encoder_state, state)
