refine python dump verilog

najaeda · Apr 4, 2024 · ed10dfb · ed10dfb
1 parent 4b98ee0
commit ed10dfb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/snl/python/snl_wrapping/PySNLDesign.cpp b/src/snl/python/snl_wrapping/PySNLDesign.cpp
@@ -102,13 +102,15 @@ static PyObject* PySNLDesign_clone(PySNLDesign* self, PyObject* args) {
 
 static PyObject* PySNLDesign_dumpVerilog(PySNLDesign* self, PyObject* args) {
   const char* arg0 = nullptr;
-  if (not PyArg_ParseTuple(args, "|s:SNLDesign.dumpVerilog", &arg0)) {
+  const char* arg1 = nullptr;
+  if (not PyArg_ParseTuple(args, "ss:SNLDesign.dumpVerilog", &arg0, &arg1)) {
     setError("malformed SNLDesign.dumpVerilog method");
     return nullptr;
   }
   METHOD_HEAD("SNLDesign.dumpVerilog()")
   SNLTRY
   SNLVRLDumper dumper;
+  dumper.setTopFileName(arg1);
   dumper.dumpDesign(selfObject, std::filesystem::path(arg0));
   SNLCATCH
   Py_RETURN_NONE;

diff --git a/test/snl/python/snl_wrapping/test_snldesign_clone2.py b/test/snl/python/snl_wrapping/test_snldesign_clone2.py
@@ -61,7 +61,7 @@ def testClone(self):
     if os.path.exists(bench_dir):
       shutil.rmtree(bench_dir)
     os.makedirs(bench_dir)
-    newDesign.dumpVerilog(os.path.join(bench_dir, bench_dir))
+    newDesign.dumpVerilog(os.path.join(bench_dir), "cloned.v")
 
     n0 = newDesign.getNet("N0")
     self.assertIsNotNone(n0)