Add test for expected fitness