I have taken some more diagnostic tests.
My bike has a torque sensor and when the power drops off it feels the same as when I stop pedalling, except it surges about every second.
Ive swapped over batteries (my wife's bike) and tried it again, still the same.
So its unlikely the battery...