@@ -207,7 +207,7 @@ def test_prepare_template_context(self, component_dir, sample_extracted_metadata
207207 assert context ['title' ] == 'Sample Component'
208208 assert context ['component_name' ] == 'sample_component'
209209
210- # Check example_code is empty string when no example_pipeline .py exists
210+ # Check example_code is empty string when no example_pipelines .py file exists
211211 assert context ['example_code' ] == ''
212212
213213 # Check formatted metadata exists
@@ -296,17 +296,17 @@ def test_load_example_pipelines(self, component_dir, sample_extracted_metadata):
296296 component_dir
297297 )
298298
299- # When no example_pipeline .py exists, should return empty string
300- example_code = generator ._load_example_pipeline ()
299+ # When no example_pipelines .py exists, should return empty string
300+ example_code = generator ._load_example_pipelines ()
301301 assert example_code == ''
302302
303- # Create an example_pipeline .py file
304- example_file = component_dir / 'example_pipeline .py'
303+ # Create an example_pipelines .py file
304+ example_file = component_dir / 'example_pipelines .py'
305305 example_content = 'from kfp import dsl\n \n @dsl.pipeline()\n def my_pipeline():\n pass'
306306 example_file .write_text (example_content )
307307
308308 # Now it should load the content
309- example_code = generator ._load_example_pipeline ()
309+ example_code = generator ._load_example_pipelines ()
310310 assert example_code == example_content
311311
312312 def test_generate_readme_component (self , component_dir , sample_extracted_metadata ):
@@ -318,18 +318,18 @@ def test_generate_readme_component(self, component_dir, sample_extracted_metadat
318318
319319 readme = generator .generate_readme ()
320320
321- # Check all sections are present (except Usage Example since no example_pipeline .py exists)
321+ # Check all sections are present (except Usage Examples since no example_pipelines .py exists)
322322 assert '# Sample Component' in readme
323323 assert '## Overview' in readme
324324 assert '## Inputs' in readme
325325 assert '## Outputs' in readme
326326 assert '## Metadata' in readme
327327
328- # Usage Example should NOT be present when example_pipeline .py doesn't exist
329- assert '## Usage Example ' not in readme
328+ # Usage Examples should NOT be present when example_pipelines .py doesn't exist
329+ assert '## Usage Examples ' not in readme
330330
331- # Now test with example_pipeline .py present
332- example_file = component_dir / 'example_pipeline .py'
331+ # Now test with example_pipelines .py present
332+ example_file = component_dir / 'example_pipelines .py'
333333 example_file .write_text ('from kfp import dsl\n \n @dsl.pipeline()\n def test_pipeline():\n pass' )
334334
335335 # Regenerate readme
@@ -339,8 +339,8 @@ def test_generate_readme_component(self, component_dir, sample_extracted_metadat
339339 )
340340 readme2 = generator2 .generate_readme ()
341341
342- # Now Usage Example should be present
343- assert '## Usage Example ' in readme2
342+ # Now Usage Examples should be present
343+ assert '## Usage Examples ' in readme2
344344 assert 'from kfp import dsl' in readme2
345345
346346 def test_generate_readme_pipeline (self , pipeline_dir , sample_extracted_metadata ):
@@ -360,8 +360,8 @@ def test_generate_readme_pipeline(self, pipeline_dir, sample_extracted_metadata)
360360 assert '## Outputs' in readme
361361 assert '## Metadata' in readme
362362
363- # Usage example should NOT be present for pipelines
364- assert '## Usage Example ' not in readme
363+ # Usage Examples should NOT be present for pipelines
364+ assert '## Usage Examples ' not in readme
365365
366366 def test_generate_readme_empty_metadata (self , temp_dir ):
367367 """Test README generation with empty metadata."""
0 commit comments