Make test docstring more generic

Optimization does not have to be about LBs
Test that unoptimizable steps are untouched

Thanks @cyli!