Added tests for generate.py

[adedamola-sode]

Created generate.py tests

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 75.68%. Comparing base (83c699b) to head (3e35186).
⚠️ Report is 1 commits behind head on screens-with-multiple-components.

Additional details and impacted files

@@                          Coverage Diff                          @@
##           screens-with-multiple-components     #157       +/-   ##
=====================================================================
+ Coverage                             55.61%   75.68%   +20.06%     
=====================================================================
  Files                                     7        7               
  Lines                                   588      588               
=====================================================================
+ Hits                                    327      445      +118     
+ Misses                                  261      143      -118     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.