move the thread-check code from train_from_dataset to a single function
add UT for the thread check function