PHP unit test - test method return generator

Question

I would like to write a unit test for method when internal MockClass will call method once and that will return Generator.

$em = $this->createMock(EntityMa