blocks_test.py 文件源码

def test_composition_forward_type_inference(self):
    b = tdb._pipe([tdb.Identity(), tdb.Identity(),
                   tdb.Map(tdb.Function(tf.negative))])
    six.assertRaisesRegex(
        self, TypeError, 'bad input type PyObjectType',
        b.input.set_input_type, tdt.PyObjectType())