increase codecov patch - add agent reporters tests

Rbnstrn · Aug 29, 2020 · 7847f00 · 7847f00
1 parent 1c07c94
commit 7847f00
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_datacollector.py b/tests/test_datacollector.py
@@ -113,15 +113,20 @@ def test_agent_records(self):
         Test agent-level variable collection.
         """
         data_collector = self.model.datacollector
+        agent_table = data_collector.get_agent_vars_dataframe()
+
         assert len(data_collector._agent_records) == 7
         for step, records in data_collector._agent_records.items():
             assert len(records) == 10
             for values in records:
                 assert len(values) == 4
 
-        assert data_collector.get_agent_vars_dataframe().shape == (70, 2)
+        assert agent_table.shape == (70, 2)
+        assert 'value' in list(agent_table.columns)
+        assert "value2" in list(agent_table.columns)
+        assert "value3" not in list(agent_table.columns)
 
-        with self.assertRaises(Exception):
+        with self.assertRaises(KeyError):
             data_collector._agent_records[8]
 
     def test_table_rows(self):