'Verifies button identified by `locator` is found from current page.
This keyword searches for buttons created with either `input` or `button` tag.
See `Page Should Contain Element` for explanation about `message` and
`loglevel` arguments.
Key attributes for buttons are `id`, `name` and `value`. See
`introduction` for details about locating elements.'

@keyword
def page_should_contain_button(self, locator, message='', loglevel='INFO'):

    try:
        self.assert_page_contains(locator, 'input', message, loglevel)
    except AssertionError:
        self.assert_page_contains(locator, 'button', message, loglevel)
