repr: introduce one_liner_max_depth=3 (closes #104)

Author: ebonnal

streamable/visitors/representation.py

@@ -32,8 +32,9 @@
 
 
 class ToStringVisitor(Visitor[str], ABC):
-    def __init__(self) -> None:
+    def __init__(self, one_liner_max_depth: int = 3) -> None:
         self.methods_reprs: List[str] = []
+        self.one_liner_max_depth = one_liner_max_depth
 
     @staticmethod
     @abstractmethod
@@ -182,10 +183,16 @@ def visit_atruncate_stream(self, stream: ATruncateStream) -> str:
         return stream.upstream.accept(self)
 
     def visit_stream(self, stream: Stream) -> str:
+        source_stream = f"Stream({self.to_string(stream.source)})"
+        depth = len(self.methods_reprs) + 1
+        if depth == 1:
+            return source_stream
+        if depth <= self.one_liner_max_depth:
+            return f"{source_stream}.{'.'.join(reversed(self.methods_reprs))}"
         methods_block = "".join(
             map(lambda r: f"    .{r}\n", reversed(self.methods_reprs))
         )
-        return f"(\n    Stream({self.to_string(stream.source)})\n{methods_block})"
+        return f"(\n    {source_stream}\n{methods_block})"
 
 
 class ReprVisitor(ToStringVisitor):

tests/test_stream.py

@@ -193,19 +193,29 @@ class CustomCallable:
         complex_stream.display(logging.ERROR)
 
         self.assertEqual(
-            """(
-    Stream(range(0, 256))
-)""",
             str(Stream(src)),
+            "Stream(range(0, 256))",
             msg="`repr` should work as expected on a stream without operation",
         )
         self.assertEqual(
+            str(Stream(src).skip(10)),
+            "Stream(range(0, 256)).skip(10, until=None)",
+            msg="`repr` should return a one-liner for a stream with 1 operations",
+        )
+        self.assertEqual(
+            str(Stream(src).skip(10).skip(10)),
+            "Stream(range(0, 256)).skip(10, until=None).skip(10, until=None)",
+            msg="`repr` should return a one-liner for a stream with 2 operations",
+        )
+        self.assertEqual(
+            str(Stream(src).skip(10).skip(10).skip(10)),
             """(
     Stream(range(0, 256))
-    .map(<lambda>, concurrency=2, ordered=True, via='process')
+    .skip(10, until=None)
+    .skip(10, until=None)
+    .skip(10, until=None)
 )""",
-            str(Stream(src).map(lambda _: _, concurrency=2, via="process")),
-            msg="`repr` should work as expected on a stream with 1 operation",
+            msg="`repr` should go to line for a stream with 3 operations",
         )
         self.assertEqual(
             str(complex_stream),