def test_stats(self): test1 = ScriptTask.from_str("python -c 'print(\"test1\")'", {'store_stdout': True}) fw = Firework(test1) self.lp.add_wf(fw) self.lp.add_wf(fw) launch_rocket(self.lp, self.fworker) launch_rocket(self.lp, self.fworker) s=FWStats(self.lp) launch_results=s.get_launch_summary(time_field="updated_on")[0] self.assertEqual((launch_results["_id"],launch_results["count"]), ("COMPLETED", 2)) self.lp.add_wf(fw) fireworks_results=s.get_fireworks_summary(time_field="updated_on") self.assertEqual((fireworks_results[1]["_id"], fireworks_results[1]["count"]), ("READY", 1)) launch_rocket(self.lp, self.fworker) workflow_results=s.get_workflow_summary(time_field="updated_on") self.assertEqual((workflow_results[0]["_id"], workflow_results[0]["count"]), ("COMPLETED", 3))