Files
galaxy/test/unit/test_rule_utils.py
T
2018-07-26 09:52:07 -04:00

34 lines
1.8 KiB
Python

import yaml
from base import rules_test_data
from galaxy.util import rules_dsl
def test_rules():
for test_case in yaml.safe_load(rules_test_data.RULES_DSL_SPEC_STR):
rule_set = rules_dsl.RuleSet(test_case)
if "initial" in test_case:
initial = test_case["initial"]
final_data, final_sources = rule_set.apply(initial["data"], initial["sources"])
expected_final = test_case["final"]
expected_final_data = expected_final["data"]
msg = "Incorrect number of rows, %s != %s" % (final_data, expected_final_data)
assert len(expected_final_data) == len(final_data), msg
for final_row, expected_final_row in zip(final_data, expected_final_data):
msg = "%s != %s" % (final_row, expected_final_row)
assert len(final_row) == len(expected_final_row), msg
for final_val, expected_final_val in zip(final_row, expected_final_row):
assert final_val == expected_final_val, msg
expected_final_sources = expected_final.get("sources", None)
if expected_final_sources:
msg = "Incorrect number of sources, %s != %s" % (expected_final_sources, final_sources)
assert len(expected_final_sources) == len(final_sources), msg
for final_source, expected_final_source in zip(final_sources, expected_final_sources):
msg = "%s != %s" % (final_source, expected_final_source)
assert final_source == expected_final_source, msg
elif "error" in test_case:
assert rule_set.has_errors, "rule [%s] does not contain errors" % test_case
else:
raise Exception("Problem with test case definition [%s]." % test_case)