From a72fadb99456b4b9097062a2e1a717d45c38a385 Mon Sep 17 00:00:00 2001
From: Tom Pollard <tpollard@mit.edu>
Date: Fri, 4 Dec 2015 12:42:11 -0500
Subject: [PATCH] add rowcount to test

---
 tests/test_build.py | 38 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 36 insertions(+), 2 deletions(-)

diff --git a/tests/test_build.py b/tests/test_build.py
index e9b1dd5..8269b8a 100644
--- a/tests/test_build.py
+++ b/tests/test_build.py
@@ -204,10 +204,44 @@ def test_itemids_in_microbiologyevents_are_in_range(self):
         """
         queryresult = pd.read_sql_query(query,self.con)
         self.assertEqual(queryresult.values[0][0],0)
+
+    # ----------------------------------------------------
+    # RUN THE FOLLOWING TESTS ON THE FULL DATASET ONLY ---
+    # ----------------------------------------------------
         
     if os.environ.has_key('USER') and os.environ['USER'] == 'jenkins':
-        def test_if_this_only_runs_in_jenkins(self):
-            self.assertEqual(os.environ['USER'],'jenkins')
+        def test_row_counts_are_as_expected(self):
+            row_dict = {
+            "ADMISSIONS": 58976,
+            "CALLOUT": 34499,
+            "CAREGIVERS": 7567,
+            "CHARTEVENTS": 263201375,
+            "CPTEVENTS": 573146,
+            "D_CPT": 134,
+            "D_ICD_DIAGNOSES": 14567,
+            "D_ICD_PROCEDURES": 3882,
+            "D_ITEMS": 12478,
+            "D_LABITEMS": 755,
+            "DATETIMEEVENTS": 4486049,
+            "DIAGNOSES_ICD": 651047,
+            "DRGCODES": 125557,
+            "ICUSTAYS": 61532,
+            "INPUTEVENTS_CV": 17528894,
+            "INPUTEVENTS_MV": 3618991,
+            "LABEVENTS": 27872575,
+            "MICROBIOLOGYEVENTS": 328446,
+            "NOTEEVENTS": 2053403,
+            "OUTPUTEVENTS": 4349339,
+            "PATIENTS": 46520,
+            "PRESCRIPTIONS": 4156848,
+            "PROCEDUREEVENTS_MV": 258066,
+            "PROCEDURES_ICD": 240095,
+            "SERVICES": 73343,
+            "TRANSFERS": 261897 }
+            for tablename,expectedrows in row_dict.iteritems():
+                query = "SELECT COUNT(*) FROM " + tablename + ";"
+                queryresult = pd.read_sql_query(query,self.con)
+                self.assertEqual(queryresult.values[0][0],expectedrows)
 
 def main():
     unittest.main()