streamlit

1
# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022-2024)
2
#
3
# Licensed under the Apache License, Version 2.0 (the "License");
4
# you may not use this file except in compliance with the License.
5
# You may obtain a copy of the License at
6
#
7
#     http://www.apache.org/licenses/LICENSE-2.0
8
#
9
# Unless required by applicable law or agreed to in writing, software
10
# distributed under the License is distributed on an "AS IS" BASIS,
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14

15
from playwright.sync_api import Page, expect
16

17
from e2e_playwright.conftest import ImageCompareFunction
18

19

20
def test_altair_chart_displays_correctly(
21
    themed_app: Page, assert_snapshot: ImageCompareFunction
22
):
23
    expect(
24
        themed_app.get_by_test_id("stArrowVegaLiteChart").locator("canvas")
25
    ).to_have_count(10)
26
    charts = themed_app.get_by_test_id("stArrowVegaLiteChart")
27
    expect(charts).to_have_count(10)
28
    snapshot_names = [
29
        "st_altair_chart-scatter_chart_default_theme",
30
        "st_altair_chart-scatter_chart_streamlit_theme",
31
        "st_altair_chart-scatter_chart_overwritten_theme",
32
        "st_altair_chart-bar_chart_overwritten_theme",
33
        "st_altair_chart-pie_chart_large_legend_items",
34
        "st_altair_chart-grouped_bar_chart_default_theme",
35
        "st_altair_chart-grouped_bar_chart_streamlit_theme",
36
        "st_altair_chart-grouped_use_container_width_default_theme"
37
        "st_altair_chart-grouped_layered_line_chart_streamlit_theme",
38
    ]
39
    for i, name in enumerate(snapshot_names):
40
        assert_snapshot(
41
            charts.nth(i),
42
            name=name,
43
        )
44