diff --git a/tests/test_interactive_image.py b/tests/test_interactive_image.py index db8b86aa9..eed252150 100644 --- a/tests/test_interactive_image.py +++ b/tests/test_interactive_image.py @@ -21,27 +21,26 @@ async def page(client: Client): with ui.tab_panel('B'): ui.label('Tab B') await client.connected() - img.set_source('https://nicegui.io/logo.png') + img.set_source('https://picsum.photos/id/29/640/360') screen.open('/') screen.wait(0.5) - assert screen.find_by_tag('img').get_attribute('src') == 'https://nicegui.io/logo.png' + assert screen.find_by_tag('img').get_attribute('src') == 'https://picsum.photos/id/29/640/360' screen.click('B') screen.wait(0.5) screen.click('A') - assert screen.find_by_tag('img').get_attribute('src') == 'https://nicegui.io/logo.png' + assert screen.find_by_tag('img').get_attribute('src') == 'https://picsum.photos/id/29/640/360' -@pytest.mark.parametrize('cross, number_of_lines', [(True, 2), (False, 0)]) -def test_with_cross(screen: Screen, cross: bool, number_of_lines: int): - ii = ui.interactive_image('https://nicegui.io/logo.png', cross=cross) - ii.content = '' +@pytest.mark.parametrize('cross', [True, False]) +def test_with_cross(screen: Screen, cross: bool): + ui.interactive_image('https://picsum.photos/id/29/640/360', + content='', cross=cross) screen.open('/') - screen.wait(0.5) screen.find_by_tag('svg') with screen.implicitly_wait(0.5): - assert len(screen.find_all_by_tag('line')) == number_of_lines + assert len(screen.find_all_by_tag('line')) == (2 if cross else 0) assert len(screen.find_all_by_tag('circle')) == 1 @@ -56,11 +55,10 @@ def replace(): ui.button('Replace', on_click=replace) screen.open('/') - screen.wait(0.5) - assert screen.find_by_tag('img').get_attribute('src').endswith('id/29/640/360') + assert (screen.find_by_tag('img').get_attribute('src') or '').endswith('id/29/640/360') screen.click('Replace') screen.wait(0.5) - assert screen.find_by_tag('img').get_attribute('src').endswith('id/30/640/360') + assert (screen.find_by_tag('img').get_attribute('src') or '').endswith('id/30/640/360') @pytest.mark.parametrize('cross', [True, False])