wrote test_em_algorithm and test_sufficient_statistics