fix readme tests

[aniketmaurya]

Before submitting

• Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
• Did you write any new necessary tests?

What does this PR do?

This PR fixes the CI for README tests which asserts the code examples in the markdown.

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in GitHub issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 89%. Comparing base (4a48ca3) to head (b5d3585).
⚠️ Report is 3 commits behind head on main.

Additional details and impacted files

@@        Coverage Diff         @@
##           main   #37   +/-   ##
==================================
  Coverage    89%   89%           
==================================
  Files         6     6           
  Lines       413   413           
==================================
  Hits        366   366           
  Misses       47    47           

🚀 New features to boost your workflow:

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