Add tests for drainExp