update test with generators