blocks_test.py 文件源码

def test_one_of(self):
    block = tdb.OneOf(lambda x: x > 0,
                      {True: tdb.Scalar(),
                       False: tdb.Scalar() >> tdb.Function(tf.negative)})
    self.assertBuildsConst(3., block, 3)
    self.assertBuildsConst(3., block, -3)