back.pysim: close .vcd/.gtkw files on context manager exit.

diff --git a/nmigen/back/pysim.py b/nmigen/back/pysim.py
index a174ec80aa6c776e181fd318132bdf573ff2809b..d6df00a8130344764a9cd182348ab1c7a4ba7550 100644 (file)
--- a/nmigen/back/pysim.py
+++ b/nmigen/back/pysim.py
@@ -579,3 +579,8 @@ class Simulator:
 
             for signal in self._gtkw_signals:
                 add_trace(signal)
+
+        if self._vcd_file:
+            self._vcd_file.close()
+        if self._gtkw_file:
+            self._gtkw_file.close()