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)
