1 files changed, 28 insertions, 0 deletions
diff --git a/src/tb/wg_tb.h b/src/tb/wg_tb.h
new file mode 100644
index 0000000..5d0a84f
--- /dev/null
+++ b/src/tb/wg_tb.h
@@ -0,0 +1,28 @@
+#pragma once
+
+#include <optical_signal.h>
+#include <systemc.h>
+#include <waveguide.h>
+#include <probe.h>
+#include <specs.h>
+
+SC_MODULE(wg_tb)
+{
+public:
+    spx::oa_port_out_type IN;
+    spx::oa_port_in_type OUT;
+
+    void run_1();
+    void monitor();
+    SC_CTOR(wg_tb)
+    {
+        SC_HAS_PROCESS(wg_tb);
+
+        SC_THREAD(run_1);
+
+        SC_THREAD(monitor);
+        sensitive << IN << OUT;
+    }
+};
+
+void wg_tb_run();