Add test for world entity optimisation