25 lines
799 B
Python
25 lines
799 B
Python
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)
|