#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# pylint: disable=E0401
"""Test Histogram module."""
import unittest
import sys
import io
from app.histogram import Histogram
from app.player import Player
[docs]class TestHistogram(unittest.TestCase):
"""Test Histogram Class."""
[docs] def test_print_histogram(self):
"""Test print_histogram by catching output and comparing it."""
histogram = Histogram()
player1 = Player("name1", False)
player2 = Player("name2", False)
res = player1.get_name()
catch_output = io.StringIO()
sys.stdout = catch_output
histogram.print_histogram(player1, player2)
sys.stdout = sys.__stdout__
output = catch_output.getvalue()
self.assertIn(str(res), output)
# check both names
self.assertIn(player1.get_name(), output)
self.assertIn(player2.get_name(), output)
[docs] def test_print_asterisk(self):
"""Test print_asterisk method."""
exp = 2
ast = ' * '
res = ' * * '
self.assertEqual(exp * ast, res)