diff --git a/poetry.lock b/poetry.lock index 6f59f04..88cec25 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1,10 +1,9 @@ -# This file is automatically @generated by Poetry 1.4.2 and should not be changed by hand. +# This file is automatically @generated by Poetry 1.8.3 and should not be changed by hand. [[package]] name = "astroid" version = "2.11.7" description = "An abstract syntax tree for Python with inference support." -category = "dev" optional = false python-versions = ">=3.6.2" files = [ @@ -23,7 +22,6 @@ wrapt = ">=1.11,<2" name = "colorama" version = "0.4.6" description = "Cross-platform colored terminal text." -category = "dev" optional = false python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,>=2.7" files = [ @@ -35,7 +33,6 @@ files = [ name = "coverage" version = "7.2.7" description = "Code coverage measurement for Python" -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -108,7 +105,6 @@ toml = ["tomli"] name = "defusedxml" version = "0.7.1" description = "XML bomb protection for Python stdlib modules" -category = "main" optional = true python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*" files = [ @@ -120,7 +116,6 @@ files = [ name = "dill" version = "0.3.7" description = "serialize all of Python" -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -135,7 +130,6 @@ graph = ["objgraph (>=1.7.2)"] name = "isort" version = "5.11.5" description = "A Python utility / library to sort Python imports." -category = "dev" optional = false python-versions = ">=3.7.0" files = [ @@ -153,7 +147,6 @@ requirements-deprecated-finder = ["pip-api", "pipreqs"] name = "joblib" version = "1.3.2" description = "Lightweight pipelining with Python functions" -category = "main" optional = false python-versions = ">=3.7" files = [ @@ -165,7 +158,6 @@ files = [ name = "lazy-object-proxy" version = "1.9.0" description = "A fast and thorough lazy object proxy." -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -211,7 +203,6 @@ files = [ name = "mccabe" version = "0.7.0" description = "McCabe checker, plugin for flake8" -category = "dev" optional = false python-versions = ">=3.6" files = [ @@ -223,7 +214,6 @@ files = [ name = "parameterized" version = "0.9.0" description = "Parameterized testing with any Python test framework" -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -238,7 +228,6 @@ dev = ["jinja2"] name = "platformdirs" version = "4.0.0" description = "A small Python package for determining appropriate platform-specific dirs, e.g. a \"user data dir\"." -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -257,7 +246,6 @@ test = ["appdirs (==1.4.4)", "covdefaults (>=2.3)", "pytest (>=7.4)", "pytest-co name = "pylint" version = "2.13.9" description = "python code static checker" -category = "dev" optional = false python-versions = ">=3.6.2" files = [ @@ -278,11 +266,70 @@ typing-extensions = {version = ">=3.10.0", markers = "python_version < \"3.10\"" [package.extras] testutil = ["gitpython (>3)"] +[[package]] +name = "pyyaml" +version = "6.0.1" +description = "YAML parser and emitter for Python" +optional = false +python-versions = ">=3.6" +files = [ + {file = "PyYAML-6.0.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:d858aa552c999bc8a8d57426ed01e40bef403cd8ccdd0fc5f6f04a00414cac2a"}, + {file = "PyYAML-6.0.1-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:fd66fc5d0da6d9815ba2cebeb4205f95818ff4b79c3ebe268e75d961704af52f"}, + {file = "PyYAML-6.0.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:69b023b2b4daa7548bcfbd4aa3da05b3a74b772db9e23b982788168117739938"}, + {file = "PyYAML-6.0.1-cp310-cp310-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:81e0b275a9ecc9c0c0c07b4b90ba548307583c125f54d5b6946cfee6360c733d"}, + {file = "PyYAML-6.0.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:ba336e390cd8e4d1739f42dfe9bb83a3cc2e80f567d8805e11b46f4a943f5515"}, + {file = "PyYAML-6.0.1-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:326c013efe8048858a6d312ddd31d56e468118ad4cdeda36c719bf5bb6192290"}, + {file = "PyYAML-6.0.1-cp310-cp310-win32.whl", hash = "sha256:bd4af7373a854424dabd882decdc5579653d7868b8fb26dc7d0e99f823aa5924"}, + {file = "PyYAML-6.0.1-cp310-cp310-win_amd64.whl", hash = "sha256:fd1592b3fdf65fff2ad0004b5e363300ef59ced41c2e6b3a99d4089fa8c5435d"}, + {file = "PyYAML-6.0.1-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:6965a7bc3cf88e5a1c3bd2e0b5c22f8d677dc88a455344035f03399034eb3007"}, + {file = "PyYAML-6.0.1-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:f003ed9ad21d6a4713f0a9b5a7a0a79e08dd0f221aff4525a2be4c346ee60aab"}, + {file = "PyYAML-6.0.1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:42f8152b8dbc4fe7d96729ec2b99c7097d656dc1213a3229ca5383f973a5ed6d"}, + {file = "PyYAML-6.0.1-cp311-cp311-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:062582fca9fabdd2c8b54a3ef1c978d786e0f6b3a1510e0ac93ef59e0ddae2bc"}, + {file = "PyYAML-6.0.1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:d2b04aac4d386b172d5b9692e2d2da8de7bfb6c387fa4f801fbf6fb2e6ba4673"}, + {file = "PyYAML-6.0.1-cp311-cp311-musllinux_1_1_x86_64.whl", hash = "sha256:e7d73685e87afe9f3b36c799222440d6cf362062f78be1013661b00c5c6f678b"}, + {file = "PyYAML-6.0.1-cp311-cp311-win32.whl", hash = "sha256:1635fd110e8d85d55237ab316b5b011de701ea0f29d07611174a1b42f1444741"}, + {file = "PyYAML-6.0.1-cp311-cp311-win_amd64.whl", hash = "sha256:bf07ee2fef7014951eeb99f56f39c9bb4af143d8aa3c21b1677805985307da34"}, + {file = "PyYAML-6.0.1-cp312-cp312-macosx_10_9_x86_64.whl", hash = "sha256:855fb52b0dc35af121542a76b9a84f8d1cd886ea97c84703eaa6d88e37a2ad28"}, + {file = "PyYAML-6.0.1-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:40df9b996c2b73138957fe23a16a4f0ba614f4c0efce1e9406a184b6d07fa3a9"}, + {file = "PyYAML-6.0.1-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a08c6f0fe150303c1c6b71ebcd7213c2858041a7e01975da3a99aed1e7a378ef"}, + {file = "PyYAML-6.0.1-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:6c22bec3fbe2524cde73d7ada88f6566758a8f7227bfbf93a408a9d86bcc12a0"}, + {file = "PyYAML-6.0.1-cp312-cp312-musllinux_1_1_x86_64.whl", hash = "sha256:8d4e9c88387b0f5c7d5f281e55304de64cf7f9c0021a3525bd3b1c542da3b0e4"}, + {file = "PyYAML-6.0.1-cp312-cp312-win32.whl", hash = "sha256:d483d2cdf104e7c9fa60c544d92981f12ad66a457afae824d146093b8c294c54"}, + {file = "PyYAML-6.0.1-cp312-cp312-win_amd64.whl", hash = "sha256:0d3304d8c0adc42be59c5f8a4d9e3d7379e6955ad754aa9d6ab7a398b59dd1df"}, + {file = "PyYAML-6.0.1-cp36-cp36m-macosx_10_9_x86_64.whl", hash = "sha256:50550eb667afee136e9a77d6dc71ae76a44df8b3e51e41b77f6de2932bfe0f47"}, + {file = "PyYAML-6.0.1-cp36-cp36m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:1fe35611261b29bd1de0070f0b2f47cb6ff71fa6595c077e42bd0c419fa27b98"}, + {file = "PyYAML-6.0.1-cp36-cp36m-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:704219a11b772aea0d8ecd7058d0082713c3562b4e271b849ad7dc4a5c90c13c"}, + {file = "PyYAML-6.0.1-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:afd7e57eddb1a54f0f1a974bc4391af8bcce0b444685d936840f125cf046d5bd"}, + {file = "PyYAML-6.0.1-cp36-cp36m-win32.whl", hash = "sha256:fca0e3a251908a499833aa292323f32437106001d436eca0e6e7833256674585"}, + {file = "PyYAML-6.0.1-cp36-cp36m-win_amd64.whl", hash = "sha256:f22ac1c3cac4dbc50079e965eba2c1058622631e526bd9afd45fedd49ba781fa"}, + {file = "PyYAML-6.0.1-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:b1275ad35a5d18c62a7220633c913e1b42d44b46ee12554e5fd39c70a243d6a3"}, + {file = "PyYAML-6.0.1-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:18aeb1bf9a78867dc38b259769503436b7c72f7a1f1f4c93ff9a17de54319b27"}, + {file = "PyYAML-6.0.1-cp37-cp37m-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:596106435fa6ad000c2991a98fa58eeb8656ef2325d7e158344fb33864ed87e3"}, + {file = "PyYAML-6.0.1-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:baa90d3f661d43131ca170712d903e6295d1f7a0f595074f151c0aed377c9b9c"}, + {file = "PyYAML-6.0.1-cp37-cp37m-win32.whl", hash = "sha256:9046c58c4395dff28dd494285c82ba00b546adfc7ef001486fbf0324bc174fba"}, + {file = "PyYAML-6.0.1-cp37-cp37m-win_amd64.whl", hash = "sha256:4fb147e7a67ef577a588a0e2c17b6db51dda102c71de36f8549b6816a96e1867"}, + {file = "PyYAML-6.0.1-cp38-cp38-macosx_10_9_x86_64.whl", hash = "sha256:1d4c7e777c441b20e32f52bd377e0c409713e8bb1386e1099c2415f26e479595"}, + {file = "PyYAML-6.0.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a0cd17c15d3bb3fa06978b4e8958dcdc6e0174ccea823003a106c7d4d7899ac5"}, + {file = "PyYAML-6.0.1-cp38-cp38-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:28c119d996beec18c05208a8bd78cbe4007878c6dd15091efb73a30e90539696"}, + {file = "PyYAML-6.0.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:7e07cbde391ba96ab58e532ff4803f79c4129397514e1413a7dc761ccd755735"}, + {file = "PyYAML-6.0.1-cp38-cp38-musllinux_1_1_x86_64.whl", hash = "sha256:49a183be227561de579b4a36efbb21b3eab9651dd81b1858589f796549873dd6"}, + {file = "PyYAML-6.0.1-cp38-cp38-win32.whl", hash = "sha256:184c5108a2aca3c5b3d3bf9395d50893a7ab82a38004c8f61c258d4428e80206"}, + {file = "PyYAML-6.0.1-cp38-cp38-win_amd64.whl", hash = "sha256:1e2722cc9fbb45d9b87631ac70924c11d3a401b2d7f410cc0e3bbf249f2dca62"}, + {file = "PyYAML-6.0.1-cp39-cp39-macosx_10_9_x86_64.whl", hash = "sha256:9eb6caa9a297fc2c2fb8862bc5370d0303ddba53ba97e71f08023b6cd73d16a8"}, + {file = "PyYAML-6.0.1-cp39-cp39-macosx_11_0_arm64.whl", hash = "sha256:c8098ddcc2a85b61647b2590f825f3db38891662cfc2fc776415143f599bb859"}, + {file = "PyYAML-6.0.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:5773183b6446b2c99bb77e77595dd486303b4faab2b086e7b17bc6bef28865f6"}, + {file = "PyYAML-6.0.1-cp39-cp39-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:b786eecbdf8499b9ca1d697215862083bd6d2a99965554781d0d8d1ad31e13a0"}, + {file = "PyYAML-6.0.1-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:bc1bf2925a1ecd43da378f4db9e4f799775d6367bdb94671027b73b393a7c42c"}, + {file = "PyYAML-6.0.1-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:04ac92ad1925b2cff1db0cfebffb6ffc43457495c9b3c39d3fcae417d7125dc5"}, + {file = "PyYAML-6.0.1-cp39-cp39-win32.whl", hash = "sha256:faca3bdcf85b2fc05d06ff3fbc1f83e1391b3e724afa3feba7d13eeab355484c"}, + {file = "PyYAML-6.0.1-cp39-cp39-win_amd64.whl", hash = "sha256:510c9deebc5c0225e8c96813043e62b680ba2f9c50a08d3724c7f28a747d1486"}, + {file = "PyYAML-6.0.1.tar.gz", hash = "sha256:bfdf460b1736c775f2ba9f6a92bca30bc2095067b8a9d77876d1fad6cc3b4a43"}, +] + [[package]] name = "setuptools" version = "68.0.0" description = "Easily download, build, install, upgrade, and uninstall Python packages" -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -299,7 +346,6 @@ testing-integration = ["build[virtualenv]", "filelock (>=3.4.0)", "jaraco.envs ( name = "tomli" version = "2.0.1" description = "A lil' TOML parser" -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -311,7 +357,6 @@ files = [ name = "typed-ast" version = "1.5.5" description = "a fork of Python 2 and 3 ast modules with type comment support" -category = "dev" optional = false python-versions = ">=3.6" files = [ @@ -362,7 +407,6 @@ files = [ name = "typing-extensions" version = "4.7.1" description = "Backported and Experimental Type Hints for Python 3.7+" -category = "dev" optional = false python-versions = ">=3.7" files = [ @@ -374,7 +418,6 @@ files = [ name = "wrapt" version = "1.16.0" description = "Module for decorators, wrappers and monkey patching." -category = "dev" optional = false python-versions = ">=3.6" files = [ @@ -451,10 +494,11 @@ files = [ ] [extras] -all = ["defusedxml"] +all = ["defusedxml", "pyyaml"] xml-loader = ["defusedxml"] +yaml-loader = ["pyyaml"] [metadata] lock-version = "2.0" python-versions = ">=3.7,<4.0" -content-hash = "6252674f8a6cc31e6860115894c00fd86d891059074bf6fdb77bbea533d9cf30" +content-hash = "c92091b8d812481714d6b5f96ced5f4fda5e024b7fc08b8c1aca9ceff68c65e7" diff --git a/pyproject.toml b/pyproject.toml index 7ae67e9..f40436c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,12 +14,14 @@ packages = [ [tool.poetry.dependencies] python = ">=3.7,<4.0" -joblib = ">=1.2,<1.4" +joblib = ">=1.2,<=1.4.2" defusedxml = { version = ">=0.7,<0.8", optional = true } +pyyaml = "^6.0.1" [tool.poetry.extras] xml_loader = ["defusedxml"] -all = ["defusedxml"] +yaml_loader = ["pyyaml"] +all = ["defusedxml", "pyyaml"] [tool.poetry.group.test.dependencies] parameterized = "*" diff --git a/pystreamapi/loaders/__init__.py b/pystreamapi/loaders/__init__.py index 877f1a8..f12d129 100644 --- a/pystreamapi/loaders/__init__.py +++ b/pystreamapi/loaders/__init__.py @@ -1,9 +1,11 @@ from pystreamapi.loaders.__csv.__csv_loader import csv from pystreamapi.loaders.__json.__json_loader import json from pystreamapi.loaders.__xml.__xml_loader import xml +from pystreamapi.loaders.__yaml.__yaml_loader import yaml __all__ = [ 'csv', 'json', - 'xml' + 'xml', + 'yaml' ] diff --git a/pystreamapi/loaders/__yaml/__init__.py b/pystreamapi/loaders/__yaml/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/pystreamapi/loaders/__yaml/__yaml_loader.py b/pystreamapi/loaders/__yaml/__yaml_loader.py new file mode 100644 index 0000000..ab88400 --- /dev/null +++ b/pystreamapi/loaders/__yaml/__yaml_loader.py @@ -0,0 +1,55 @@ +try: + import yaml as yaml_lib +except ImportError as exc: + raise ImportError( + "Please install the yaml_loader extra dependency to use the yaml loader." + ) from exc +from collections import namedtuple + +from pystreamapi.loaders.__lazy_file_iterable import LazyFileIterable +from pystreamapi.loaders.__loader_utils import LoaderUtils + + +def yaml(src: str, read_from_src=False) -> LazyFileIterable: + """ + Loads YAML data from either a path or a string and converts it into a list of namedtuples. + + Args: + src (str): Either the path to a YAML file or a YAML string. + read_from_src (bool): If True, src is treated as a YAML string. If False, src is treated as + a path to a YAML file. + + Returns: + list: A list of namedtuples, where each namedtuple represents an object in the YAML. + """ + if read_from_src: + return LazyFileIterable(lambda: __load_yaml_string(src)) + path = LoaderUtils.validate_path(src) + return LazyFileIterable(lambda: __load_yaml_file(path)) + + +def __load_yaml_file(file_path): + """Load a YAML file and convert it into a list of namedtuples""" + with open(file_path, mode='r', encoding='utf-8') as yamlfile: + src = yamlfile.read() + if src == '': + return [] + data = yaml_lib.safe_load(src) + return __convert_to_namedtuples(data) + + +def __load_yaml_string(yaml_string): + """Load YAML data from a string and convert it into a list of namedtuples""" + data = yaml_lib.safe_load(yaml_string) + return [] if data is None else __convert_to_namedtuples(data) + + +def __convert_to_namedtuples(data, name='Item'): + """Convert YAML data to a list of namedtuples""" + if isinstance(data, dict): + fields = list(data.keys()) + Item = namedtuple(name, fields) + return Item(**{k: __convert_to_namedtuples(v, k) for k, v in data.items()}) + if isinstance(data, list): + return [__convert_to_namedtuples(item, name) for item in data] + return data diff --git a/tests/_loaders/test_json_loader.py b/tests/_loaders/test_json_loader.py index 44404b5..20cd044 100644 --- a/tests/_loaders/test_json_loader.py +++ b/tests/_loaders/test_json_loader.py @@ -8,18 +8,23 @@ file_content = """ [ - { - "attr1": 1, - "attr2": 2.0 - }, - { - "attr1": "a", - "attr2": "b" - } + { + "attr1": 1, + "attr2": 2.0 + }, + { + "attr1": [ + { + "attr1": "a" + } + ], + "attr2": "b" + } ] """ file_path = 'path/to/data.json' + class TestJsonLoader(TestCase): def test_json_loader_from_file(self): @@ -27,13 +32,7 @@ def test_json_loader_from_file(self): patch(PATH_EXISTS, return_value=True), patch(PATH_ISFILE, return_value=True)): data = json(file_path) - self.assertEqual(len(data), 2) - self.assertEqual(data[0].attr1, 1) - self.assertIsInstance(data[0].attr1, int) - self.assertEqual(data[0].attr2, 2.0) - self.assertIsInstance(data[0].attr2, float) - self.assertEqual(data[1].attr1, 'a') - self.assertIsInstance(data[1].attr1, str) + self._check_extracted_data(data) def test_json_loader_is_iterable(self): with (patch(OPEN, mock_open(read_data=file_content)), @@ -59,14 +58,17 @@ def test_json_loader_with_no_file(self): def test_json_loader_from_string(self): data = json(file_content, read_from_src=True) + self._check_extracted_data(data) + + def test_json_loader_from_empty_string(self): + with self.assertRaises(JSONDecodeError): + len(json('', read_from_src=True)) + + def _check_extracted_data(self, data): self.assertEqual(len(data), 2) self.assertEqual(data[0].attr1, 1) self.assertIsInstance(data[0].attr1, int) self.assertEqual(data[0].attr2, 2.0) self.assertIsInstance(data[0].attr2, float) - self.assertEqual(data[1].attr1, 'a') - self.assertIsInstance(data[1].attr1, str) - - def test_json_loader_from_empty_string(self): - with self.assertRaises(JSONDecodeError): - len(json('', read_from_src=True)) + self.assertIsInstance(data[1].attr1, list) + self.assertEqual(data[1].attr1[0].attr1, 'a') diff --git a/tests/_loaders/test_yaml_loader.py b/tests/_loaders/test_yaml_loader.py new file mode 100644 index 0000000..f9beee1 --- /dev/null +++ b/tests/_loaders/test_yaml_loader.py @@ -0,0 +1,64 @@ +# pylint: disable=not-context-manager +from unittest import TestCase +from unittest.mock import patch, mock_open + +from _loaders.file_test import OPEN, PATH_EXISTS, PATH_ISFILE +from pystreamapi.loaders import yaml + +file_content = """ +--- +- attr1: 1 + attr2: 2.0 +- attr1: + - attr1: a + attr2: b +""" +file_path = 'path/to/data.yaml' + + +class TestYamlLoader(TestCase): + + def test_yaml_loader_from_file(self): + with (patch(OPEN, mock_open(read_data=file_content)), + patch(PATH_EXISTS, return_value=True), + patch(PATH_ISFILE, return_value=True)): + data = yaml(file_path) + self._check_extracted_data(data) + + def test_yaml_loader_is_iterable(self): + with (patch(OPEN, mock_open(read_data=file_content)), + patch(PATH_EXISTS, return_value=True), + patch(PATH_ISFILE, return_value=True)): + data = yaml(file_path) + self.assertEqual(len(list(iter(data))), 2) + + def test_yaml_loader_with_empty_file(self): + with (patch(OPEN, mock_open(read_data="")), + patch(PATH_EXISTS, return_value=True), + patch(PATH_ISFILE, return_value=True)): + data = yaml(file_path) + self.assertEqual(len(data), 0) + + def test_yaml_loader_with_invalid_path(self): + with self.assertRaises(FileNotFoundError): + yaml('path/to/invalid.yaml') + + def test_yaml_loader_with_no_file(self): + with self.assertRaises(ValueError): + yaml('../') + + def test_yaml_loader_from_string(self): + data = yaml(file_content, read_from_src=True) + self._check_extracted_data(data) + + def test_yaml_loader_from_empty_string(self): + self.assertEqual(list(yaml('', read_from_src=True)), []) + + def _check_extracted_data(self, data): + self.assertEqual(len(data), 2) + self.assertEqual(data[0].attr1, 1) + self.assertIsInstance(data[0].attr1, int) + self.assertEqual(data[0].attr2, 2.0) + self.assertIsInstance(data[0].attr2, float) + self.assertIsInstance(data[1].attr1, list) + self.assertEqual(data[1].attr1[0].attr1, 'a')