Unit Tests Revisit with Model Update

[edwinnglabs]

With all the model updated from previous version, consider including most of the args into the unit tests such as hyper-parameters tuning.

Also need to report a coverage % in the badge somewhere.