def test_visualizing_array_node_with_filter(): x_node = Array() x_node.addfilter(Missing("p", "missing_value")) tree = x_node.get_parametrized_tree({'p': 0.5, 'missing_value': np.nan}) path = plotting_utils.visualize_error_generator(tree, False) file = open(path, 'r') data = file.read() assert re.compile(r'2.*Missing.*probability: 0').search(data) assert re.compile(r'1 -> 2').search(data)
def test_visualizing_array_node_with_complex_filter(): x_node = Array() const = Constant("c") addition = Addition(const, const) x_node.addfilter(addition) path = plotting_utils.visualize_error_generator( x_node.get_parametrized_tree({'c': 5}), False) file = open(path, 'r') data = file.read() assert re.compile(r'2.*Addition').search(data) assert re.compile(r'3.*Constant.*value: 5').search(data) assert re.compile(r'4.*Constant.*value: 5').search(data) assert re.compile(r'2 -> 3.*filter_a').search(data) assert re.compile(r'2 -> 4.*filter_b').search(data)