|
2 | 2 |
|
3 | 3 | from flatten_array import flatten |
4 | 4 |
|
5 | | - |
6 | 5 | # Tests adapted from `problem-specifications//canonical-data.json` @ v1.2.0 |
7 | 6 |
|
8 | | -class FlattenArrayTest(unittest.TestCase): |
9 | 7 |
|
| 8 | +class FlattenArrayTest(unittest.TestCase): |
10 | 9 | def test_no_nesting(self): |
11 | | - self.assertEqual(flatten([0, 1, 2]), [0, 1, 2]) |
| 10 | + inputs = [0, 1, 2] |
| 11 | + expected = [0, 1, 2] |
| 12 | + self.assertEqual(flatten(inputs), expected) |
12 | 13 |
|
13 | | - def test_flatten_integers(self): |
| 14 | + def test_flattens_array_with_just_integers_present(self): |
14 | 15 | inputs = [1, [2, 3, 4, 5, 6, 7], 8] |
15 | 16 | expected = [1, 2, 3, 4, 5, 6, 7, 8] |
16 | 17 | self.assertEqual(flatten(inputs), expected) |
17 | 18 |
|
18 | | - def test_five_level_nesting(self): |
| 19 | + def test_5_level_nesting(self): |
19 | 20 | inputs = [0, 2, [[2, 3], 8, 100, 4, [[[50]]]], -2] |
20 | 21 | expected = [0, 2, 2, 3, 8, 100, 4, 50, -2] |
21 | 22 | self.assertEqual(flatten(inputs), expected) |
22 | 23 |
|
23 | | - def test_six_level_nesting(self): |
| 24 | + def test_6_level_nesting(self): |
24 | 25 | inputs = [1, [2, [[3]], [4, [[5]]], 6, 7], 8] |
25 | 26 | expected = [1, 2, 3, 4, 5, 6, 7, 8] |
26 | 27 | self.assertEqual(flatten(inputs), expected) |
27 | 28 |
|
28 | | - def test_with_none_values(self): |
| 29 | + def test_6_level_nest_list_with_null_values(self): |
29 | 30 | inputs = [0, 2, [[2, 3], 8, [[100]], None, [[None]]], -2] |
30 | 31 | expected = [0, 2, 2, 3, 8, 100, -2] |
31 | 32 | self.assertEqual(flatten(inputs), expected) |
32 | 33 |
|
33 | | - def test_all_values_are_none(self): |
| 34 | + def test_all_values_in_nested_list_are_null(self): |
34 | 35 | inputs = [None, [[[None]]], None, None, [[None, None], None], None] |
35 | 36 | expected = [] |
36 | 37 | self.assertEqual(flatten(inputs), expected) |
37 | 38 |
|
38 | | - # Additional tests for this track |
39 | | - def test_empty_nested_lists(self): |
40 | | - self.assertEqual(flatten([[()]]), []) |
41 | | - |
42 | | - def test_strings(self): |
43 | | - self.assertEqual(flatten(['0', ['1', '2']]), ['0', '1', '2']) |
44 | | - |
45 | 39 |
|
46 | | -if __name__ == '__main__': |
| 40 | +if __name__ == "__main__": |
47 | 41 | unittest.main() |
0 commit comments