streamlit

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

17
describe("st._arrow_scatter_chart", () => {
18
  before(() => {
19
    cy.loadApp("http://localhost:3000/");
20

21
    cy.prepForElementSnapshots();
22
  });
23

24
  it("displays a scatterplot chart", () => {
25
    cy.get(".element-container [data-testid='stArrowVegaLiteChart']")
26
      .find("canvas")
27
      .first()
28
      .should("have.css", "height", "350px");
29
  });
30

31
  it("displays all scatterplot-chart combinations correctly", () => {
32
    cy.get(".element-container [data-testid='stArrowVegaLiteChart']").should(
33
      "have.length",
34
      5
35
    );
36

37
    const getChart = (index) => (
38
      cy.get(".element-container [data-testid='stArrowVegaLiteChart']").eq(index)
39
    );
40

41
    getChart(0).matchThemedSnapshots("arrow_scatter_chart_snowpark");
42
    getChart(1).matchThemedSnapshots("arrow_scatter_chart_no_data");
43
    getChart(2).matchThemedSnapshots("arrow_scatter_chart_align_color_and_few_sizes");
44
    getChart(3).matchThemedSnapshots("arrow_scatter_chart_align_color_and_many_sizes");
45
    getChart(4).matchThemedSnapshots("arrow_scatter_chart_align_quant_color_and_size");
46
  });
47
});
48