... However ... if you don't have wifi set in advanced settings to "Keep Wifi on when screen is off" then when the phone sleeps wifi will be disabled and use mobile data for notifications. And, on the same screen in advanced settings you should see an option to "switch to mobile data" when the internet is not available on WiFi, so your phone will chose mobile data, even if WiFi is on, if the WiFi signal drops below a certain threshold.