fix: test

Author: clatapie

src/pyconverter/xml2py/ast_tree.py

@@ -3020,8 +3020,6 @@ def py_docstring(self, custom_functions: CustomFunctions, warning_command_dict:
         """
         xml_cmd = f"{self._terms['pn006p']} Command: `{self.name} <{self.url}>`_"
 
-        # ``warning_command_dict`` need to be added here
-
         items = [self.short_desc, "", xml_cmd]
 
         if self.name in warning_command_dict.keys():

src/pyconverter/xml2py/load_xml_doc.py

@@ -109,7 +109,7 @@ def grab_links(linkmap):
                 if targetptr is not None and href is not None:
                     text = ""
                     if linkmap[0].tag == "ttl":
-                        text = str(linkmap[0])
+                        text = str(linkmap[0]).strip()
                     links[f"{targetptr}"] = (root_name, root_title, href, text)
 
         grab_links(linkmap)

tests/conftest.py

@@ -26,7 +26,7 @@
 from pyconverter.xml2py.custom_functions import CustomFunctions
 import pyconverter.xml2py.directory_format as ff
 import pyconverter.xml2py.load_xml_doc as lxd
-from pyconverter.xml2py.utils.utils import get_config_data_value
+from pyconverter.xml2py.utils.utils import get_config_data_value, get_warning_command_dict
 import pyconverter.xml2py.writer as wrt
 import pytest
 
@@ -142,3 +142,8 @@ def library_name_structured(config_path):
 @pytest.fixture
 def package_structure(command_map, name_map, directory_path, cwd, path_custom_functions):
     return wrt.write_source(command_map, name_map, directory_path, cwd, path_custom_functions)
+
+
+@pytest.fixture
+def warning_command_dict(config_path):
+    return get_warning_command_dict(config_path)

tests/test_writer.py

@@ -26,7 +26,7 @@
 import pytest
 
 
-def test_convert(command_map, custom_functions):
+def test_convert(command_map, custom_functions, warning_command_dict):
     assert command_map["/XFRM"].name == "/XFRM"
     assert (
         "Original command: WRITE\n\nShort Description:\nWrites the radiation matrix file.\n\nFunction signature:\nWRITE,"  # noqa : E501
@@ -39,9 +39,9 @@ def test_convert(command_map, custom_functions):
     assert 'def zoom(self, wn: str = "", lab: str = "", x1: str = "", y1: str = "", x2: str = "", y2: str = "", **kwargs):\n    r"""Zooms a region of a display window.\n\n' in command_map[  # noqa : E501
         "/ZOOM"
     ].to_python(
-        custom_functions
+        custom_functions, warning_command_dict
     )
-    assert "import re" in command_map["K"].to_python(custom_functions)
+    assert "import re" in command_map["K"].to_python(custom_functions, warning_command_dict)
 
 
 def test_copy_template_package(cwd):

tests/test_xml_ast.py

@@ -27,7 +27,7 @@
 
 @pytest.mark.parametrize(
     "initial_name,expected_output",
-    [("/PREP7", "prep7"), ("*DMAT", "dmat"), ("SORT", "sort"), ("*GO", "stargo")],
+    [("/PREP7", "prep7"), ("*DMAT", "dmat"), ("SORT", "sort"), ("*VPLOT", "starvplot")],
 )
 def test_py_name(initial_name, expected_output, name_map):
     assert ast.to_py_name(initial_name, name_map) == expected_output