def test_restrict_outputs():
"""Test a basic use of the restrict(outputs=(...)) method."""
raise SkipTest
code = 'x = a + b\ny = b - c\nz = c**2'
b = Block(code)
br = b.restrict(outputs=('z',))
names = dict(c=5)
br.execute(names)
assert_equal(sorted(names), ['c', 'z'])
assert_equal(names['c'], 5)
assert_equal(names['z'], 25)
def test_restrict_outputs():
"""Test a basic use of the restrict(outputs=(...)) method."""
raise SkipTest
code = 'x = a + b\ny = b - c\nz = c**2'
b = Block(code)
br = b.restrict(outputs=('z', ))
names = dict(c=5)
br.execute(names)
assert_equal(sorted(names), ['c', 'z'])
assert_equal(names['c'], 5)
assert_equal(names['z'], 25)
def test_restrict_inputs():
"""Test a basic use of the restrict(inputs=(...)) method."""
raise SkipTest
code = 'x = a + b\ny = b - c\nz = c**2'
b = Block(code)
assert_equal(b.inputs, set(['a','b','c']))
assert_equal(b.outputs, set(['x','y','z']))
br = b.restrict(inputs=('a',))
names = dict(a=100, b=200)
br.execute(names)
assert_equal(sorted(names), ['a', 'b', 'x'])
assert_equal(names['a'], 100)
assert_equal(names['b'], 200)
assert_equal(names['x'], 300)
def test_restrict_inputs():
"""Test a basic use of the restrict(inputs=(...)) method."""
raise SkipTest
code = 'x = a + b\ny = b - c\nz = c**2'
b = Block(code)
assert_equal(b.inputs, set(['a', 'b', 'c']))
assert_equal(b.outputs, set(['x', 'y', 'z']))
br = b.restrict(inputs=('a', ))
names = dict(a=100, b=200)
br.execute(names)
assert_equal(sorted(names), ['a', 'b', 'x'])
assert_equal(names['a'], 100)
assert_equal(names['b'], 200)
assert_equal(names['x'], 300)
