Adding tests directory into machine learning(using pytest) #1607