test_variable.py 文件源码

def test_constructor(self):
        # verify allowed_dims
        for allowed_dims in (tuple(), list(), ''):
            var = Variable(allowed_dims)
            assert var.allowed_dims == ((),)

        for allowed_dims in ('x', ['x'], ('x')):
            var = Variable(allowed_dims)
            assert var.allowed_dims == (('x',),)

        var = Variable(('x', 'y'))
        assert var.allowed_dims == (('x', 'y'),)

        var = Variable([(), 'x', ('x', 'y')])
        assert var.allowed_dims == ((), ('x',), ('x', 'y'))