Corey Morris
commited on
Commit
·
02b1702
1
Parent(s):
cd21f99
added failing test for new behavior of organization column. Updated test for rows for the newly added rows
Browse files
test_result_data_processing.py
CHANGED
|
@@ -23,7 +23,7 @@ class TestResultDataProcessor(unittest.TestCase):
|
|
| 23 |
# check that the number of rows is correct
|
| 24 |
def test_rows(self):
|
| 25 |
data = self.processor.data
|
| 26 |
-
self.assertEqual(len(data),
|
| 27 |
|
| 28 |
# check that mc1 column exists
|
| 29 |
def test_mc1(self):
|
|
@@ -42,5 +42,11 @@ class TestResultDataProcessor(unittest.TestCase):
|
|
| 42 |
self.assertLess(mc1.max(), 1.0)
|
| 43 |
self.assertGreater(mc1.min(), 0.0)
|
| 44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
if __name__ == '__main__':
|
| 46 |
unittest.main()
|
|
|
|
| 23 |
# check that the number of rows is correct
|
| 24 |
def test_rows(self):
|
| 25 |
data = self.processor.data
|
| 26 |
+
self.assertEqual(len(data), 998)
|
| 27 |
|
| 28 |
# check that mc1 column exists
|
| 29 |
def test_mc1(self):
|
|
|
|
| 42 |
self.assertLess(mc1.max(), 1.0)
|
| 43 |
self.assertGreater(mc1.min(), 0.0)
|
| 44 |
|
| 45 |
+
|
| 46 |
+
# test that a column named organization exists
|
| 47 |
+
def test_organization(self):
|
| 48 |
+
data = self.processor.data
|
| 49 |
+
self.assertIn('organization', data.columns)
|
| 50 |
+
|
| 51 |
if __name__ == '__main__':
|
| 52 |
unittest.main()
|