Generalized WaterBox test to allow other water models to be specified.