# test_stats.py
# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
#
# This module is part of GitPython and is released under
# the BSD License: http://www.opensource.org/licenses/bsd-license.php

from test.lib import TestBase, fixture
from git import Stats
from git.compat import defenc


class TestStats(TestBase):
    def test_list_from_string(self):
        output = fixture("diff_numstat").decode(defenc)
        stats = Stats._list_from_string(self.rorepo, output)

        self.assertEqual(2, stats.total["files"])
        self.assertEqual(52, stats.total["lines"])
        self.assertEqual(29, stats.total["insertions"])
        self.assertEqual(23, stats.total["deletions"])

        self.assertEqual(29, stats.files["a.txt"]["insertions"])
        self.assertEqual(18, stats.files["a.txt"]["deletions"])

        self.assertEqual(0, stats.files["b.txt"]["insertions"])
        self.assertEqual(5, stats.files["b.txt"]["deletions"])