Skip to content

Commit 9d89efc

Browse files
AiryShiftsmalley
authored andcommitted
flatten-array: update tests to v1.1.0 (exercism#1046)
* flatten-array: update tests to v1.1.0
1 parent 866e0eb commit 9d89efc

File tree

1 file changed

+18
-11
lines changed

1 file changed

+18
-11
lines changed

exercises/flatten-array/flatten_array_test.py

Lines changed: 18 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,35 +3,42 @@
33
from flatten_array import flatten
44

55

6+
# Tests adapted from `problem-specifications//canonical-data.json` @ v1.1.0
7+
68
class FlattenArrayTests(unittest.TestCase):
79

810
def test_no_nesting(self):
911
self.assertEqual(flatten([0, 1, 2]), [0, 1, 2])
1012

11-
def test_one_level_nesting(self):
12-
self.assertEqual(flatten([0, [1], 2]), [0, 1, 2])
13-
14-
def test_two_level_nesting(self):
15-
self.assertEqual(flatten([0, [1, [2, 3]], [4]]), [0, 1, 2, 3, 4])
16-
17-
def test_empty_nested_lists(self):
18-
self.assertEqual(flatten([[()]]), [])
13+
def test_flatten_integers(self):
14+
inputs = [1, [2, 3, 4, 5, 6, 7], 8]
15+
expected = [1, 2, 3, 4, 5, 6, 7, 8]
16+
self.assertEqual(flatten(inputs), expected)
1917

20-
def test_with_none_values(self):
21-
inputs = [0, 2, [[2, 3], 8, [[100]], None, [[None]]], -2]
22-
expected = [0, 2, 2, 3, 8, 100, -2]
18+
def test_five_level_nesting(self):
19+
inputs = [0, 2, [[2, 3], 8, 100, 4, [[[50]]]], -2]
20+
expected = [0, 2, 2, 3, 8, 100, 4, 50, -2]
2321
self.assertEqual(flatten(inputs), expected)
2422

2523
def test_six_level_nesting(self):
2624
inputs = [1, [2, [[3]], [4, [[5]]], 6, 7], 8]
2725
expected = [1, 2, 3, 4, 5, 6, 7, 8]
2826
self.assertEqual(flatten(inputs), expected)
2927

28+
def test_with_none_values(self):
29+
inputs = [0, 2, [[2, 3], 8, [[100]], None, [[None]]], -2]
30+
expected = [0, 2, 2, 3, 8, 100, -2]
31+
self.assertEqual(flatten(inputs), expected)
32+
3033
def test_all_values_are_none(self):
3134
inputs = [None, [[[None]]], None, None, [[None, None], None], None]
3235
expected = []
3336
self.assertEqual(flatten(inputs), expected)
3437

38+
# Additional tests for this track
39+
def test_empty_nested_lists(self):
40+
self.assertEqual(flatten([[()]]), [])
41+
3542
def test_strings(self):
3643
self.assertEqual(flatten(['0', ['1', '2']]), ['0', '1', '2'])
3744

0 commit comments

Comments
 (0)