def test_register_yaml_constructor(figure_yaml, expected_figure, yaml_loader):
class TestLoader(yaml_loader):
pass
Figure.register_yaml_constructor(TestLoader, tag='!fig')
generated_figure = yaml.load(figure_yaml, Loader=TestLoader)
assert expected_figure == generated_figure
assert isinstance(generated_figure.value, type(expected_figure.value))
