blocks_test.py 文件源码

def test_one_of_mixed_input_type(self):
    block = (tdb.Identity(), tdb.Scalar('int32')) >> tdb.OneOf(
        key_fn=tdb.GetItem(0),
        case_blocks=(tdb.Function(tf.square), tdb.Function(tf.negative)),
        pre_block=tdb.GetItem(1))
    self.assertBuilds(4, block, (0, 2))
    self.assertBuilds(-2, block, (1, 2))