def test_equal(self): parser = Parser( "value(test_unit_type.unit.conversion(product=p_1))", "value(test_unit_type.unit.conversion(product=p_1))", ) assert len(parser.get_goals()) == 1 parser = Parser( "value(test_unit_type.unit.conversion(product=p_1))", "value(test_unit_type.unit.conversion)", ) assert len(parser.get_goals()) == 2 parser = Parser( "value(test_unit_type.unit.conversion(product=p_1)) + value(test_unit_type.unit.conversion(product=p_1))", "value(test_unit_type.unit.conversion)", ) assert len(parser.get_goals()) == 2
def test_get_goals_dimensional(self): parser = Parser( 'count(test_unit_type.global.conversion(product=p_1)) + count(test_unit_type.global.conversion)', 'count(test_unit_type.unit.conversion(product=p_1_2))', ) goals = parser.get_goals() dim_goals = [g for g in goals if g.is_dimensional()] assert len(goals) == 3 assert len(dim_goals) == 2 assert set([dg.dimension for dg in dim_goals]) == {'product'} assert set([dg.dimension_value for dg in dim_goals]) == {'p_1', 'p_1_2'}