From da334e47aa32ea8e016ba76330053787094a99ea Mon Sep 17 00:00:00 2001
From: mrg <mrg@ucsc.edu>
Date: Wed, 1 Apr 2020 11:14:50 -0700
Subject: [PATCH] Separate pbitcell tests for precharge

---
 compiler/tests/04_precharge_pbitcell_test.py | 52 ++++++++++++++++++++
 compiler/tests/04_precharge_test.py          | 22 +--------
 2 files changed, 53 insertions(+), 21 deletions(-)
 create mode 100755 compiler/tests/04_precharge_pbitcell_test.py

diff --git a/compiler/tests/04_precharge_pbitcell_test.py b/compiler/tests/04_precharge_pbitcell_test.py
new file mode 100755
index 00000000..fd2f8283
--- /dev/null
+++ b/compiler/tests/04_precharge_pbitcell_test.py
@@ -0,0 +1,52 @@
+#!/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
+
+class precharge_test(openram_test):
+
+    def runTest(self):
+        config_file = "{}/tests/configs/config".format(os.getenv("OPENRAM_HOME"))
+        globals.init_openram(config_file)
+        
+        # check precharge in multi-port
+        OPTS.bitcell = "pbitcell"
+        OPTS.num_rw_ports = 1
+        OPTS.num_r_ports = 1
+        OPTS.num_w_ports = 1
+
+        factory.reset()
+        debug.info(2, "Checking precharge for pbitcell (innermost connections)")
+        tx = factory.create(module_type="precharge", size=1, bitcell_bl="bl0", bitcell_br="br0")
+        self.local_check(tx)
+
+        factory.reset()
+        debug.info(2, "Checking precharge for pbitcell (innermost connections)")
+        tx = factory.create(module_type="precharge", size=1, bitcell_bl="bl1", bitcell_br="br1")
+        self.local_check(tx)
+        
+        factory.reset()
+        debug.info(2, "Checking precharge for pbitcell (outermost connections)")
+        tx = factory.create(module_type="precharge", size=1, bitcell_bl="bl2", bitcell_br="br2")
+        self.local_check(tx)
+
+        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())
diff --git a/compiler/tests/04_precharge_test.py b/compiler/tests/04_precharge_test.py
index 1c12ad8b..76d433a8 100755
--- a/compiler/tests/04_precharge_test.py
+++ b/compiler/tests/04_precharge_test.py
@@ -15,6 +15,7 @@ from globals import OPTS
 from sram_factory import factory
 import debug
 
+
 class precharge_test(openram_test):
 
     def runTest(self):
@@ -26,27 +27,6 @@ class precharge_test(openram_test):
         tx = factory.create(module_type="precharge", size=1)
         self.local_check(tx)
         
-        # check precharge in multi-port
-        OPTS.bitcell = "pbitcell"
-        OPTS.num_rw_ports = 1
-        OPTS.num_r_ports = 1
-        OPTS.num_w_ports = 1
-
-        factory.reset()
-        debug.info(2, "Checking precharge for pbitcell (innermost connections)")
-        tx = factory.create(module_type="precharge", size=1, bitcell_bl="bl0", bitcell_br="br0")
-        self.local_check(tx)
-
-        factory.reset()
-        debug.info(2, "Checking precharge for pbitcell (innermost connections)")
-        tx = factory.create(module_type="precharge", size=1, bitcell_bl="bl1", bitcell_br="br1")
-        self.local_check(tx)
-        
-        factory.reset()
-        debug.info(2, "Checking precharge for pbitcell (outermost connections)")
-        tx = factory.create(module_type="precharge", size=1, bitcell_bl="bl2", bitcell_br="br2")
-        self.local_check(tx)
-
         globals.end_openram()
         
 # run the test from the command line