diff --git a/src/pystencilssfg/lang/headers.py b/src/pystencilssfg/lang/headers.py index fdb8f17be3ddbe177994c67099cebeaeb719b67e..550dd4d4c17e4d8bc6d3bfb83e498ce41f32eb14 100644 --- a/src/pystencilssfg/lang/headers.py +++ b/src/pystencilssfg/lang/headers.py @@ -24,6 +24,9 @@ class HeaderFile: return header system_header = False + if header.startswith('"') and header.endswith('"'): + header = header[1:-1] + if header.startswith("<") and header.endswith(">"): header = header[1:-1] system_header = True diff --git a/tests/lang/test_headers.py b/tests/lang/test_headers.py new file mode 100644 index 0000000000000000000000000000000000000000..3eaf040bc159f91556abb039d921ba81ee0228fe --- /dev/null +++ b/tests/lang/test_headers.py @@ -0,0 +1,13 @@ +from pystencilssfg.lang import HeaderFile +import pytest + + +def test_parse_system(): + headerfile = HeaderFile.parse("<test>") + assert str(headerfile) == "<test>" and headerfile.system_header + + +@pytest.mark.parametrize("header_string", ["test.hpp", '"test.hpp"']) +def test_parse_private(header_string): + headerfile = HeaderFile.parse(header_string) + assert str(headerfile) == "test.hpp" and not headerfile.system_header