From a8df5528f9ddf2c044735e3cae75df8fb85b6a63 Mon Sep 17 00:00:00 2001
From: jsowash <jsowash@vpn-pool-128-114-231-151.ucsc.edu>
Date: Wed, 21 Aug 2019 16:06:36 -0700
Subject: [PATCH] Added 2 mux test for wmask.

---
 .../tests/20_sram_1bank_2mux_wmask_test.py    | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 compiler/tests/20_sram_1bank_2mux_wmask_test.py

diff --git a/compiler/tests/20_sram_1bank_2mux_wmask_test.py b/compiler/tests/20_sram_1bank_2mux_wmask_test.py
new file mode 100644
index 00000000..65f025a7
--- /dev/null
+++ b/compiler/tests/20_sram_1bank_2mux_wmask_test.py
@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+# See LICENSE for licensing information.
+#
+# Copyright (c) 2016-2019 Regents of the University of California and The Board
+# of Regents for the Oklahoma Agricultural and Mechanical College
+# (acting for and on behalf of Oklahoma State University)
+# All rights reserved.
+#
+import unittest
+from testutils import *
+import sys, os
+
+sys.path.append(os.getenv("OPENRAM_HOME"))
+import globals
+from globals import OPTS
+from sram_factory import factory
+import debug
+
+
+# @unittest.skip("SKIPPING 20_sram_1bank_2mux_wmask_test")
+class sram_1bank_2mux_wmask_test(openram_test):
+
+    def runTest(self):
+        globals.init_openram("config_{0}".format(OPTS.tech_name))
+        from sram_config import sram_config
+        c = sram_config(word_size=8,
+                        write_size=4,
+                        num_words=64,
+                        num_banks=1)
+
+        c.words_per_row = 2
+        c.recompute_sizes()
+        debug.info(1, "Layout test for {}rw,{}r,{}w sram "
+                      "with {} bit words, {} words, {} bit writes, {} words per "
+                      "row, {} banks".format(OPTS.num_rw_ports,
+                                             OPTS.num_r_ports,
+                                             OPTS.num_w_ports,
+                                             c.word_size,
+                                             c.num_words,
+                                             c.write_size,
+                                             c.words_per_row,
+                                             c.num_banks))
+        a = factory.create(module_type="sram", sram_config=c)
+        self.local_check(a, final_verification=True)
+
+        globals.end_openram()
+
+
+# run the test from the command line
+if __name__ == "__main__":
+    (OPTS, args) = globals.parse_args()
+    del sys.argv[1:]
+    header(__file__, OPTS.tech_name)
+    unittest.main(testRunner=debugTestRunner())
\ No newline at end of file