Yep absolutely works as well. Was just on my phone and gave an example.

But arduino = 5v logic, and esp = 3.3v logic.

Also ESP does have a lot higher base load than a atmega chip. You can do tricks to get those down to microwatts.