added test for 1 epoch