import unittest
from utility import visualization


class NumRowsTestCase(unittest.TestCase):
    def test_get_num_rows(self):
        """Test case for number of row calculation."""

        num_rows1 = visualization._get_num_rows(num_graphs=4, num_cols=3)
        expected_rows1 = 2

        num_rows2 = visualization._get_num_rows(num_graphs=5, num_cols=2)
        expected_rows2 = 3

        num_rows3 = visualization._get_num_rows(num_graphs=4, num_cols=4)
        expected_rows3 = 1

        num_rows4 = visualization._get_num_rows(num_graphs=4, num_cols=5)
        expected_rows4 = 1

        self.assertEqual(num_rows1, expected_rows1)
        self.assertEqual(num_rows2, expected_rows2)
        self.assertEqual(num_rows3, expected_rows3)
        self.assertEqual(num_rows4, expected_rows4)