Add test for non-linear pricing

Testing non-linear pricing requires a fair bit of set-up.