Ch131 (Python): Task 1

3 files changed, 68 insertions, 0 deletions

diff --git a/challenge-131/tyler-wardhaugh/python/ch-1.py b/challenge-131/tyler-wardhaugh/python/ch-1.py
new file mode 100755
index 0000000000..b59141f114
--- /dev/null
+++ b/challenge-131/tyler-wardhaugh/python/ch-1.py
@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+"""Challenge 131, Task 1"""
+
+import sys
+
+
+DEFAULT_INPUT = [[1, 2, 3, 6, 7, 8, 9]]
+
+
+def consecutive_arrays(coll: list[int]) -> list[list[int]]:
+    if not coll:
+        return [[]]
+
+    results = []
+    prev = coll[0]
+    for v in coll:
+        if prev + 1 != v:
+            results.append([])
+        results[-1].append(v)
+        prev = v
+        
+    return results
+
+
+def main(args: list[str] = None) -> None:
+    """Run the task"""
+    if args is None:
+        args = sys.argv[1:]
+
+    coll = None
+    if args:
+        import json
+        coll = json.loads(args[0])
+    else:
+        coll = DEFAULT_INPUT[0]
+
+    results = consecutive_arrays(coll)
+    print(f'({str(results)[1:-1]})')
+
+
+if __name__ == '__main__':
+    sys.exit(main())
diff --git a/challenge-131/tyler-wardhaugh/python/ch1.py b/challenge-131/tyler-wardhaugh/python/ch1.py
new file mode 120000
index 0000000000..7680b02e4f
--- /dev/null
+++ b/challenge-131/tyler-wardhaugh/python/ch1.py
@@ -0,0 +1 @@
+ch-1.py
\ No newline at end of file
diff --git a/challenge-131/tyler-wardhaugh/python/test_ch1.py b/challenge-131/tyler-wardhaugh/python/test_ch1.py
new file mode 100755
index 0000000000..87a0e71ef8
--- /dev/null
+++ b/challenge-131/tyler-wardhaugh/python/test_ch1.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+"""Test Task 1"""
+
+import unittest
+from ch1 import consecutive_arrays
+
+
+class TestTask1(unittest.TestCase):
+    """Test Task 1"""
+
+    def test_example_cases(self):
+        """Test Task 1"""
+        self.assertEqual([[1, 2, 3], [6, 7, 8, 9]],
+                consecutive_arrays([1, 2, 3, 6, 7, 8, 9]))
+        self.assertEqual([[11, 12], [14], [17, 18, 19]],
+                consecutive_arrays([11, 12, 14, 17, 18, 19]))
+        self.assertEqual([[2], [4], [6], [8]],
+                consecutive_arrays([2, 4, 6, 8]))
+        self.assertEqual([[1, 2, 3, 4, 5]],
+                consecutive_arrays([1, 2, 3, 4, 5]))
+
+
+
+if __name__ == '__main__':
+    unittest.main()