Add pytest for ModParse

Author: rodrigomelo9

hdltools/mod_parse.py

@@ -15,7 +15,7 @@
 from hdltools.hdl_sanitize import HdlSanitize
 
 
-class Modparse:
+class ModParse:
     """Extract information about parameters and ports from modules."""
 
     def __init__(self, fpath):
@@ -101,5 +101,5 @@ def __str__(self):
     parser = argparse.ArgumentParser()
     parser.add_argument('svfile')
     args = parser.parse_args()
-    modules = Modparse(args.svfile)
+    modules = ModParse(args.svfile)
     print(modules)

tests/test_modparse.py

@@ -0,0 +1,24 @@
+from pathlib import Path
+
+from hdltools.mod_parse import ModParse
+
+vfile = Path(__file__).parent.resolve() / 'hdl/modules.sv'
+
+
+def test_modules():
+    vobj = ModParse(vfile)
+    modules = vobj.get_modules()
+    assert len(modules) == 3
+
+
+def test_params():
+    vobj = ModParse(vfile)
+    modules = vobj.get_modules()
+    assert 'params' in modules['mod_param']
+
+
+def test_ports():
+    vobj = ModParse(vfile)
+    modules = vobj.get_modules()
+    assert 'ports' in modules['mod_param']
+    assert 'ports' in modules['mod_nopar']