diff --git a/src/timeseries/meta.py b/src/timeseries/meta.py index 0412685..a7c5b46 100644 --- a/src/timeseries/meta.py +++ b/src/timeseries/meta.py @@ -120,7 +120,7 @@ def __eq__(self, other) -> bool: return trees_equal and entities_equal - def tree(self, *args, **kwargs) -> str: + def print_tree(self, *args, **kwargs) -> str: # ugly! it would be preferable not to print the tree to std out # ... but this works import io diff --git a/tests/timeseries/test_meta.py b/tests/timeseries/test_meta.py index cb9bc08..46ce3bb 100644 --- a/tests/timeseries/test_meta.py +++ b/tests/timeseries/test_meta.py @@ -130,7 +130,7 @@ def test_read_flat_codes_from_klass() -> None: @log_start_stop def test_read_hierarchical_codes_from_klass() -> None: energy_balance = Taxonomy(157) - ts_logger.debug(f"captured ...\n{energy_balance.tree(attr_list=['fullName'])}") + ts_logger.debug(f"captured ...\n{energy_balance.print_tree()}") assert energy_balance.structure.root.name == "0" assert energy_balance.structure.max_depth == 4 @@ -217,8 +217,7 @@ def test_hierarchical_codes_retrieved_from_klass_and_reloaded_from_json_file_are finally: os.remove(temp_file) - file157 - # compare for leaf nodes of sub tree + # compare all leaf nodes of sub tree k157_names = [n.name for n in klass157.structure.root.leaves] f157_names = [n.name for n in file157.structure.root.leaves] assert k157_names == f157_names