Enrich string examples

README.md

@@ -50,6 +50,23 @@ Program weight: 3
 Program return type: int
 ```
 
+We could also try a more involved example in the string domain:
+```
+python synthesis.py --domain strings --example concatenate_3 --max-weight 5 
+
+Synthesis Log:
+- Extracted 13 constants from examples.
+- Searching level 2 with 13 primitives.
+- Searching level 3 with 79 primitives.
+- Searching level 4 with 79 primitives.
+
+Synthesis Results:
+- Program found in 1.9864s.
+- Program: Concat(x0, Concat(x1, x2))
+- Program weight: 5
+- Program return type: str
+```
+
 To add additional input-output examples, modify `examples.py`. Add a new key to the dictionary `example_set` and set the value to be a list of tuples.
 
 ## 🔎 Abstract Syntax Tree
@@ -85,7 +102,7 @@ source setup_jupyter.sh
 
 ## 🙏🏽 Acknowledgements
 
-I thank [Tyler Holloway](mailto:[email protected]), teaching fellow in CS252R, for her guidance in completing this implementation of bottom-up enumerative program synthesis.
+I thank [Tyler Holloway](mailto:[email protected]), teaching fellow in CS252R, for her guidance.
 
 ## 📫 Contact
 

examples.py

@@ -17,12 +17,17 @@ example_set = {
     'multiplication': [([2, 3], 6), ([4, 5], 20), ([7, 8], 56), ([9, 2], 18), ([3, 4], 12)],
     'division': [([6, 2], 3), ([8, 4], 2), ([9, 3], 3), ([10, 5], 2), ([12, 6], 2)],
 
-    # complex arithmetic examples
+    # advanced arithmetic examples
     'add_5_multiply_2': [([1, 2], 12), ([3, 4], 22), ([5, 6], 32), ([7, 8], 42), ([9, 10], 52)],
     'multiply_add_9': [([1, 2], 11), ([3, 4], 21), ([5, 6], 39), ([7, 8], 65), ([9, 10], 9)],
 
-    # string examples
+    # basic string examples
     'concatenate': [(["a", "b"], "ab"), (["c", "d"], "cd"), (["e", "f"], "ef")],
+    'right': [(["hello", 3], "llo"), (["world", 4], "orld"), (["fox", 1], "x")],
+    'left': [(["hello", 2], "he"), (["world", 3], "wor"), (["fox", 2], "fo")],
+
+    # advanced string examples
+    'concatenate_3': [(["a", "b", "c"], "abc"), (["d", "e", "f"], "def"), (["g", "h", "i"], "ghi")],
 
     # custom user examples
 }

synthesis.py

@@ -30,7 +30,7 @@ def parse_args():
     parser = argparse.ArgumentParser(description="Bottom-up enumerative synthesis in Python.")
 
     # define valid choices for the 'domain' argument
-    valid_domain_choices = ["arithmetic", "string"]
+    valid_domain_choices = ["arithmetic", "strings"]
 
     # add examples
     parser.add_argument('--domain', type=str, required=True, # default="arithmetic",
@@ -139,7 +139,7 @@ def run_synthesizer(args):
     # define operators
     if args.domain == "arithmetic":
         operators = arithmetic_operators
-    elif args.domain == "string":
+    elif args.domain == "strings":
         operators = string_operators
     else:
         raise Exception('Domain not recognized. Must be either "arithmetic" or "string".')