change lifespan test to get rid of the deprecation warning.

edwenger · May 14, 2018 · 642fd17 · 642fd17
1 parent c6a51fa
commit 642fd17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_lifespan.py b/tests/test_lifespan.py
@@ -81,7 +81,7 @@ def test_ticks_seen(self):
         assert self.df.steps.max() == 1
 
     def test_agent_lifetime(self):
-        lifetimes = self.df.groupby(["AgentID"]).Step.agg({"Step":
+        lifetimes = self.df.groupby(["AgentID"]).agg({"Step":
                                                            lambda x: len(x)})
         assert lifetimes.Step.max() == 2