modified the unit test for sample_from_gaussian