def test_get_identifiers_recursively(): ''' Test finding identifiers including subexpressions. ''' variables = {} variables['sub1'] = Subexpression(Unit(1), np.float32, 'sub2 * z', variables, {}) variables['sub2'] = Subexpression(Unit(1), np.float32, '5 + y', variables, {}) variables['x'] = Variable(unit=None) identifiers = get_identifiers_recursively('_x = sub1 + x', variables) assert identifiers == set(['x', '_x', 'y', 'z', 'sub1', 'sub2'])
def test_get_identifiers_recursively(): ''' Test finding identifiers including subexpressions. ''' variables = {'sub1': Subexpression(name='sub1', unit=Unit(1), dtype=np.float32, expr='sub2 * z', owner=FakeGroup(variables={}), device=None), 'sub2': Subexpression(name='sub2', unit=Unit(1), dtype=np.float32, expr='5 + y', owner=FakeGroup(variables={}), device=None), 'x': Variable(unit=None, name='x')} identifiers = get_identifiers_recursively('_x = sub1 + x', variables) assert identifiers == set(['x', '_x', 'y', 'z', 'sub1', 'sub2'])
def test_get_identifiers_recursively(): ''' Test finding identifiers including subexpressions. ''' variables = {'sub1': Subexpression(name='sub1', dtype=np.float32, expr='sub2 * z', owner=FakeGroup(variables={}), device=None), 'sub2': Subexpression(name='sub2', dtype=np.float32, expr='5 + y', owner=FakeGroup(variables={}), device=None), 'x': Variable(name='x')} identifiers = get_identifiers_recursively(['_x = sub1 + x'], variables) assert identifiers == {'x', '_x', 'y', 'z', 'sub1', 'sub2'}