test: rename column names for birth_names dataset (#12082)

This commit is contained in:
Jesse Yang 2021-01-08 14:13:20 -08:00 committed by GitHub
parent f354bb3d1b
commit 2210015aab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
11 changed files with 55 additions and 55 deletions

View File

@ -27,7 +27,7 @@ describe('AdhocMetrics', () => {
}); });
it('Clear metric and set simple adhoc metric', () => { it('Clear metric and set simple adhoc metric', () => {
const metric = 'sum(sum_girls)'; const metric = 'sum(num_girls)';
const metricName = 'Sum Girls'; const metricName = 'Sum Girls';
cy.get('[data-test=metrics]') cy.get('[data-test=metrics]')
.find('[data-test="remove-control-button"]') .find('[data-test="remove-control-button"]')
@ -40,7 +40,7 @@ describe('AdhocMetrics', () => {
cy.get('[data-test="AdhocMetricEditTitle#trigger"]').click(); cy.get('[data-test="AdhocMetricEditTitle#trigger"]').click();
cy.get('[data-test="AdhocMetricEditTitle#input"]').type(metricName); cy.get('[data-test="AdhocMetricEditTitle#input"]').type(metricName);
cy.get('[name="select-column"]').click().type('sum_girls{enter}'); cy.get('[name="select-column"]').click().type('num_girls{enter}');
cy.get('[name="select-aggregate"]').click().type('sum{enter}'); cy.get('[name="select-aggregate"]').click().type('sum{enter}');
cy.get('[data-test="AdhocMetricEdit#save"]').contains('Save').click(); cy.get('[data-test="AdhocMetricEdit#save"]').contains('Save').click();
@ -94,7 +94,7 @@ describe('AdhocMetrics', () => {
xit('Switch from custom sql tabs to simple', () => { xit('Switch from custom sql tabs to simple', () => {
cy.get('[data-test=metrics]').within(() => { cy.get('[data-test=metrics]').within(() => {
cy.get('.Select__dropdown-indicator').click(); cy.get('.Select__dropdown-indicator').click();
cy.get('input[type=text]').type('sum_girls{enter}'); cy.get('input[type=text]').type('num_girls{enter}');
}); });
cy.get('[data-test=metrics]') cy.get('[data-test=metrics]')
.find('[data-test="metric-option"]') .find('[data-test="metric-option"]')
@ -102,7 +102,7 @@ describe('AdhocMetrics', () => {
cy.get('#metrics-edit-popover').within(() => { cy.get('#metrics-edit-popover').within(() => {
cy.get('#adhoc-metric-edit-tabs-tab-SQL').click(); cy.get('#adhoc-metric-edit-tabs-tab-SQL').click();
cy.get('.ace_identifier').contains('sum_girls'); cy.get('.ace_identifier').contains('num_girls');
cy.get('.ace_content').click(); cy.get('.ace_content').click();
cy.get('.ace_text-input').type('{selectall}{backspace}SUM(num)'); cy.get('.ace_text-input').type('{selectall}{backspace}SUM(num)');
cy.get('#adhoc-metric-edit-tabs-tab-SIMPLE').click(); cy.get('#adhoc-metric-edit-tabs-tab-SIMPLE').click();

View File

@ -39,17 +39,17 @@ describe('Visualization > Pivot Table', () => {
expressionType: 'SIMPLE', expressionType: 'SIMPLE',
column: { column: {
id: 338, id: 338,
column_name: 'sum_boys', column_name: 'num_boys',
expression: '', expression: '',
filterable: false, filterable: false,
groupby: false, groupby: false,
is_dttm: false, is_dttm: false,
type: 'BIGINT', type: 'BIGINT',
optionName: '_col_sum_boys', optionName: '_col_num_boys',
}, },
aggregate: 'SUM', aggregate: 'SUM',
hasCustomLabel: false, hasCustomLabel: false,
label: 'SUM(sum_boys)', label: 'SUM(num_boys)',
optionName: 'metric_gvpdjt0v2qf_6hkf56o012', optionName: 'metric_gvpdjt0v2qf_6hkf56o012',
}; };
@ -88,7 +88,7 @@ describe('Visualization > Pivot Table', () => {
metrics: ['sum__num', TEST_METRIC], metrics: ['sum__num', TEST_METRIC],
}); });
cy.get('.chart-container tr:eq(0) th:eq(1)').contains('sum__num'); cy.get('.chart-container tr:eq(0) th:eq(1)').contains('sum__num');
cy.get('.chart-container tr:eq(0) th:eq(2)').contains('SUM(sum_boys)'); cy.get('.chart-container tr:eq(0) th:eq(2)').contains('SUM(num_boys)');
cy.get('.chart-container tr:eq(1) th:eq(0)').contains('state'); cy.get('.chart-container tr:eq(1) th:eq(0)').contains('state');
cy.get('.chart-container tr:eq(2) th:eq(0)').contains('name'); cy.get('.chart-container tr:eq(2) th:eq(0)').contains('name');
}); });
@ -100,7 +100,7 @@ describe('Visualization > Pivot Table', () => {
metrics: ['sum__num', TEST_METRIC], metrics: ['sum__num', TEST_METRIC],
}); });
cy.get('.chart-container tr:eq(0) th:eq(2)').contains('sum__num'); cy.get('.chart-container tr:eq(0) th:eq(2)').contains('sum__num');
cy.get('.chart-container tr:eq(0) th:eq(3)').contains('SUM(sum_boys)'); cy.get('.chart-container tr:eq(0) th:eq(3)').contains('SUM(num_boys)');
cy.get('.chart-container tr:eq(2) th:eq(0)').contains('name'); cy.get('.chart-container tr:eq(2) th:eq(0)').contains('name');
cy.get('.chart-container tr:eq(2) th:eq(1)').contains('gender'); cy.get('.chart-container tr:eq(2) th:eq(1)').contains('gender');
}); });

View File

@ -31,7 +31,7 @@ describe('Visualization > Table', () => {
const PERCENT_METRIC = { const PERCENT_METRIC = {
expressionType: 'SQL', expressionType: 'SQL',
sqlExpression: 'CAST(SUM(sum_girls)+AS+FLOAT)/SUM(num)', sqlExpression: 'CAST(SUM(num_girls)+AS+FLOAT)/SUM(num)',
column: null, column: null,
aggregate: null, aggregate: null,
hasCustomLabel: true, hasCustomLabel: true,

View File

@ -24,19 +24,19 @@ export default {
verbose_map: { verbose_map: {
count: 'COUNT(*)', count: 'COUNT(*)',
__timestamp: 'Time', __timestamp: 'Time',
sum__sum_girls: 'sum__sum_girls', sum__num_girls: 'sum__num_girls',
name: 'name', name: 'name',
avg__sum_girls: 'avg__sum_girls', avg__num_girls: 'avg__num_girls',
gender: 'gender', gender: 'gender',
sum_girls: 'sum_girls', num_girls: 'num_girls',
ds: 'ds', ds: 'ds',
sum__sum_boys: 'sum__sum_boys', sum__num_boys: 'sum__num_boys',
state: 'state', state: 'state',
num: 'num', num: 'num',
sum__num: 'sum__num', sum__num: 'sum__num',
sum_boys: 'sum_boys', num_boys: 'num_boys',
avg__num: 'avg__num', avg__num: 'avg__num',
avg__sum_boys: 'avg__sum_boys', avg__num_boys: 'avg__num_boys',
}, },
metrics: [ metrics: [
{ {
@ -54,31 +54,31 @@ export default {
description: null, description: null,
}, },
{ {
expression: 'SUM(birth_names.sum_boys)', expression: 'SUM(birth_names.num_boys)',
warning_text: null, warning_text: null,
verbose_name: 'sum__sum_boys', verbose_name: 'sum__num_boys',
metric_name: 'sum__sum_boys', metric_name: 'sum__num_boys',
description: null, description: null,
}, },
{ {
expression: 'AVG(birth_names.sum_boys)', expression: 'AVG(birth_names.num_boys)',
warning_text: null, warning_text: null,
verbose_name: 'avg__sum_boys', verbose_name: 'avg__num_boys',
metric_name: 'avg__sum_boys', metric_name: 'avg__num_boys',
description: null, description: null,
}, },
{ {
expression: 'SUM(birth_names.sum_girls)', expression: 'SUM(birth_names.num_girls)',
warning_text: null, warning_text: null,
verbose_name: 'sum__sum_girls', verbose_name: 'sum__num_girls',
metric_name: 'sum__sum_girls', metric_name: 'sum__num_girls',
description: null, description: null,
}, },
{ {
expression: 'AVG(birth_names.sum_girls)', expression: 'AVG(birth_names.num_girls)',
warning_text: null, warning_text: null,
verbose_name: 'avg__sum_girls', verbose_name: 'avg__num_girls',
metric_name: 'avg__sum_girls', metric_name: 'avg__num_girls',
description: null, description: null,
}, },
{ {
@ -149,7 +149,7 @@ export default {
is_dttm: false, is_dttm: false,
expression: '', expression: '',
groupby: false, groupby: false,
column_name: 'sum_boys', column_name: 'num_boys',
}, },
{ {
type: 'BIGINT', type: 'BIGINT',
@ -159,7 +159,7 @@ export default {
is_dttm: false, is_dttm: false,
expression: '', expression: '',
groupby: false, groupby: false,
column_name: 'sum_girls', column_name: 'num_girls',
}, },
], ],
id, id,
@ -189,10 +189,10 @@ export default {
['["num", false]', 'num [desc]'], ['["num", false]', 'num [desc]'],
['["state", true]', 'state [asc]'], ['["state", true]', 'state [asc]'],
['["state", false]', 'state [desc]'], ['["state", false]', 'state [desc]'],
['["sum_boys", true]', 'sum_boys [asc]'], ['["num_boys", true]', 'num_boys [asc]'],
['["sum_boys", false]', 'sum_boys [desc]'], ['["num_boys", false]', 'num_boys [desc]'],
['["sum_girls", true]', 'sum_girls [asc]'], ['["num_girls", true]', 'num_girls [asc]'],
['["sum_girls", false]', 'sum_girls [desc]'], ['["num_girls", false]', 'num_girls [desc]'],
], ],
type: 'table', type: 'table',
edit_url: '/tablemodelview/edit/7', edit_url: '/tablemodelview/edit/7',

View File

@ -63,7 +63,7 @@ def gen_filter(
def load_data(tbl_name: str, database: Database, sample: bool = False) -> None: def load_data(tbl_name: str, database: Database, sample: bool = False) -> None:
pdf = pd.read_json(get_example_data("birth_names.json.gz")) pdf = pd.read_json(get_example_data("birth_names2.json.gz"))
# TODO(bkyryliuk): move load examples data into the pytest fixture # TODO(bkyryliuk): move load examples data into the pytest fixture
if database.backend == "presto": if database.backend == "presto":
pdf.ds = pd.to_datetime(pdf.ds, unit="ms") pdf.ds = pd.to_datetime(pdf.ds, unit="ms")
@ -221,14 +221,14 @@ def create_slices(tbl: BaseDatasource) -> Tuple[List[Slice], List[Slice]]:
metrics=[ metrics=[
{ {
"expressionType": "SIMPLE", "expressionType": "SIMPLE",
"column": {"column_name": "sum_boys", "type": "BIGINT(20)"}, "column": {"column_name": "num_boys", "type": "BIGINT(20)"},
"aggregate": "SUM", "aggregate": "SUM",
"label": "Boys", "label": "Boys",
"optionName": "metric_11", "optionName": "metric_11",
}, },
{ {
"expressionType": "SIMPLE", "expressionType": "SIMPLE",
"column": {"column_name": "sum_girls", "type": "BIGINT(20)"}, "column": {"column_name": "num_girls", "type": "BIGINT(20)"},
"aggregate": "SUM", "aggregate": "SUM",
"label": "Girls", "label": "Girls",
"optionName": "metric_12", "optionName": "metric_12",

View File

@ -106,7 +106,7 @@ class TestExportDatabasesCommand(SupersetTestCase):
"metric_name": "ratio", "metric_name": "ratio",
"verbose_name": "Ratio Boys/Girls", "verbose_name": "Ratio Boys/Girls",
"metric_type": None, "metric_type": None,
"expression": "sum(sum_boys) / sum(sum_girls)", "expression": "sum(num_boys) / sum(num_girls)",
"description": "This represents the ratio of boys/girls", "description": "This represents the ratio of boys/girls",
"d3format": ".2%", "d3format": ".2%",
"extra": None, "extra": None,
@ -159,7 +159,7 @@ class TestExportDatabasesCommand(SupersetTestCase):
"python_date_format": None, "python_date_format": None,
}, },
{ {
"column_name": "sum_girls", "column_name": "num_girls",
"verbose_name": None, "verbose_name": None,
"is_dttm": False, "is_dttm": False,
"is_active": None, "is_active": None,
@ -195,7 +195,7 @@ class TestExportDatabasesCommand(SupersetTestCase):
"python_date_format": None, "python_date_format": None,
}, },
{ {
"column_name": "sum_boys", "column_name": "num_boys",
"verbose_name": None, "verbose_name": None,
"is_dttm": None, "is_dttm": None,
"is_active": None, "is_active": None,

View File

@ -240,8 +240,8 @@ class TestImportDatasetsCommand(SupersetTestCase):
"state", "state",
"gender", "gender",
"name", "name",
"sum_boys", "num_boys",
"sum_girls", "num_girls",
"num", "num",
] ]
@ -276,8 +276,8 @@ class TestImportDatasetsCommand(SupersetTestCase):
"state", "state",
"gender", "gender",
"name", "name",
"sum_boys", "num_boys",
"sum_girls", "num_girls",
"num", "num",
} }

View File

@ -34,7 +34,7 @@ class TestDatasource(SupersetTestCase):
resp = self.get_json_resp(url) resp = self.get_json_resp(url)
col_names = {o.get("name") for o in resp} col_names = {o.get("name") for o in resp}
self.assertEqual( self.assertEqual(
col_names, {"sum_boys", "num", "gender", "name", "ds", "state", "sum_girls"} col_names, {"num_boys", "num", "gender", "name", "ds", "state", "num_girls"}
) )
def test_external_metadata_for_virtual_table(self): def test_external_metadata_for_virtual_table(self):
@ -179,13 +179,13 @@ class TestDatasource(SupersetTestCase):
self.assertEqual( self.assertEqual(
col_names, col_names,
{ {
"sum_boys", "num_boys",
"num", "num",
"gender", "gender",
"name", "name",
"ds", "ds",
"state", "state",
"sum_girls", "num_girls",
"num_california", "num_california",
}, },
) )

View File

@ -77,7 +77,7 @@ datasource_post = {
}, },
{ {
"id": 509, "id": 509,
"column_name": "sum_boys", "column_name": "num_boys",
"verbose_name": None, "verbose_name": None,
"description": None, "description": None,
"expression": None, "expression": None,
@ -88,7 +88,7 @@ datasource_post = {
}, },
{ {
"id": 510, "id": 510,
"column_name": "sum_girls", "column_name": "num_girls",
"verbose_name": None, "verbose_name": None,
"description": None, "description": None,
"expression": "", "expression": "",
@ -144,7 +144,7 @@ datasource_post = {
"metric_name": "ratio", "metric_name": "ratio",
"verbose_name": "Ratio Boys/Girls", "verbose_name": "Ratio Boys/Girls",
"description": "This represents the ratio of boys/girls", "description": "This represents the ratio of boys/girls",
"expression": "sum(sum_boys) / sum(sum_girls)", "expression": "sum(num_boys) / sum(num_girls)",
"warning_text": "no warning", "warning_text": "no warning",
"d3format": ".2%", "d3format": ".2%",
}, },

View File

@ -30,8 +30,8 @@ dataset_ui_export: List[Dict[str, Any]] = [
{"column_name": "state", "type": "VARCHAR(10)"}, {"column_name": "state", "type": "VARCHAR(10)"},
{"column_name": "gender", "type": "VARCHAR(16)"}, {"column_name": "gender", "type": "VARCHAR(16)"},
{"column_name": "name", "type": "VARCHAR(255)"}, {"column_name": "name", "type": "VARCHAR(255)"},
{"column_name": "sum_boys", "type": "BIGINT"}, {"column_name": "num_boys", "type": "BIGINT"},
{"column_name": "sum_girls", "type": "BIGINT"}, {"column_name": "num_girls", "type": "BIGINT"},
{"column_name": "num", "type": "BIGINT"}, {"column_name": "num", "type": "BIGINT"},
], ],
"filter_select_enabled": True, "filter_select_enabled": True,
@ -201,7 +201,7 @@ dashboard_export: Dict[str, Any] = {
"__TableColumn__": { "__TableColumn__": {
"changed_by_fk": None, "changed_by_fk": None,
"changed_on": {"__datetime__": "2020-10-07T15:50:00"}, "changed_on": {"__datetime__": "2020-10-07T15:50:00"},
"column_name": "sum_boys", "column_name": "num_boys",
"created_by_fk": None, "created_by_fk": None,
"created_on": {"__datetime__": "2020-10-07T15:50:00"}, "created_on": {"__datetime__": "2020-10-07T15:50:00"},
"description": None, "description": None,
@ -222,7 +222,7 @@ dashboard_export: Dict[str, Any] = {
"__TableColumn__": { "__TableColumn__": {
"changed_by_fk": None, "changed_by_fk": None,
"changed_on": {"__datetime__": "2020-10-07T15:50:00"}, "changed_on": {"__datetime__": "2020-10-07T15:50:00"},
"column_name": "sum_girls", "column_name": "num_girls",
"created_by_fk": None, "created_by_fk": None,
"created_on": {"__datetime__": "2020-10-07T15:50:00"}, "created_on": {"__datetime__": "2020-10-07T15:50:00"},
"description": None, "description": None,

View File

@ -271,7 +271,7 @@ class TestSqlaTableModel(SupersetTestCase):
spec.allows_joins = inner_join spec.allows_joins = inner_join
arbitrary_gby = "state || gender || '_test'" arbitrary_gby = "state || gender || '_test'"
arbitrary_metric = dict( arbitrary_metric = dict(
label="arbitrary", expressionType="SQL", sqlExpression="SUM(sum_boys)" label="arbitrary", expressionType="SQL", sqlExpression="SUM(num_boys)"
) )
query_obj = dict( query_obj = dict(
groupby=[arbitrary_gby, "name"], groupby=[arbitrary_gby, "name"],