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)