add expected blocks

Author: reeba212

sdks/python/apache_beam/yaml/examples/bigtable_enrichment.yaml

@@ -48,5 +48,8 @@ pipeline:
   # We can also store it somewhere like a json file
     - type: LogForTesting
 
-  # Expected:
-  #  Row(sale_id = 1, customer_id = 1, product_id = 1, quantity = 1, product = Row(product_id: "1", product_name = "pixel 5", product_stock = "2"))
+options:
+  yaml_experimental_features: Enrichment
+  
+# Expected:
+#  Row(sale_id=1, customer_id=1, product_id=1, quantity=1, product={'product_id': '1', 'product_name': 'pixel 5', 'product_stock': '2'})

sdks/python/apache_beam/yaml/examples/enrich_spanner_with_bigquery.yaml

@@ -94,3 +94,6 @@ pipeline:
       config:
         path: 'errors.json'
 
+# Expected:
+#  Row(customer_id=1001, customer_name='Alice', customer_email='[email protected]', product_id=2001, order_date='24-03-24', order_amount=150)
+#  Row(customer_id=1003, customer_name='Claire', customer_email='[email protected]', product_id=2003, order_date='7-05-24', order_amount=110)

sdks/python/apache_beam/yaml/examples/testing/examples_test.py

@@ -155,8 +155,14 @@ def _wordcount_test_preprocessor(
       env.input_file('kinglear.txt', '\n'.join(lines)))
 
 
-@YamlExamplesTestSuite.register_test_preprocessor(
-    ['test_simple_filter_yaml', 'test_simple_filter_and_combine_yaml', 'test_spanner_read_yaml', 'test_spanner_write_yaml'])
+@YamlExamplesTestSuite.register_test_preprocessor([
+    'test_simple_filter_yaml',
+    'test_simple_filter_and_combine_yaml',
+    'test_spanner_read_yaml',
+    'test_spanner_write_yaml',
+    'test_bigtable_enrichment_yaml',
+    'test_enrich_spanner_with_bigquery_yaml'
+])
 def _file_io_write_test_preprocessor(
     test_spec: dict, expected: List[str], env: TestEnvironment):