# 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'])