Fixed bugs in visualization.show_BER_curves

This commit is contained in:
2022-11-11 17:38:15 +01:00
parent 6105eef4c1
commit 0f8de32e3f
2 changed files with 7 additions and 6 deletions

View File

@@ -6,16 +6,16 @@ class NumRowsTestCase(unittest.TestCase):
def test_get_num_rows(self):
"""Test case for number of row calculation."""
num_rows1 = visualization._get_num_rows(4, 2)
num_rows1 = visualization._get_num_rows(num_graphs=4, num_cols=3)
expected_rows1 = 2
num_rows2 = visualization._get_num_rows(5, 2)
num_rows2 = visualization._get_num_rows(num_graphs=5, num_cols=2)
expected_rows2 = 3
num_rows3 = visualization._get_num_rows(4, 4)
num_rows3 = visualization._get_num_rows(num_graphs=4, num_cols=4)
expected_rows3 = 1
num_rows4 = visualization._get_num_rows(4, 5)
num_rows4 = visualization._get_num_rows(num_graphs=4, num_cols=5)
expected_rows4 = 1
self.assertEqual(num_rows1, expected_rows1)