def METHOD_NAME(self):
    df = pd.DataFrame(
        data=[[1, 15, 5, 20], [1, 234, 45, 6]],
        columns=["taskId", "1_count", "2_count", "3_count"],
    )
    result = calc_count(df)
    self.assertEqual(result[0], 40)