Add unit test for client training

[moreyhat]

Is your feature request related to a problem? Please describe.
No, it would be just for keeping the code quality.

Describe the solution you'd like
Add unit test for client training.

Describe alternatives you've considered
N/A

Additional context
N/A