import pytest import yaml def test_dump(): assert yaml.dump(['foo']) def test_load_no_loader(): with pytest.raises(TypeError): yaml.load("- foo\n") def test_load_safeloader(): assert yaml.load("- foo\n", Loader=yaml.SafeLoader)