- add test for train examples

[huangshiyu13]

Description

Types of changes

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)

Checklist

• I have ensured make test pass (required).
• I have checked the code using make commit-checks (required).

[codecov]

Codecov Report

Merging #40 (0c7fb57) into main (64209f8) will increase coverage by 6.01%.
The diff coverage is n/a.

@@            Coverage Diff             @@
##             main      #40      +/-   ##
==========================================
+ Coverage   35.74%   41.76%   +6.01%     
==========================================
  Files         115      115              
  Lines        7179     7179              
==========================================
+ Hits         2566     2998     +432     
+ Misses       4613     4181     -432     

Flag	Coverage Δ
unittests	41.76% <ø> (+6.01%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

see 20 files with indirect coverage changes

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more