fix: Bump prophet, re-enable tests, and remedy column eligibility logic (#24129)

This commit is contained in:
John Bodley
2023-07-05 12:33:52 -07:00
committed by GitHub
parent 0836000f9f
commit 383dac6a5e
6 changed files with 35 additions and 12 deletions

View File

@@ -444,11 +444,11 @@ class TestPostChartDataApi(BaseTestChartDataApi):
else:
raise Exception("ds column not found")
@pytest.mark.usefixtures("load_birth_names_dashboard_with_slices")
def test_chart_data_prophet(self):
"""
Chart data API: Ensure prophet post transformation works
"""
pytest.importorskip("prophet")
time_grain = "P1Y"
self.query_context_payload["queries"][0]["is_timeseries"] = True
self.query_context_payload["queries"][0]["groupby"] = []
@@ -476,7 +476,7 @@ class TestPostChartDataApi(BaseTestChartDataApi):
self.assertIn("sum__num__yhat", row)
self.assertIn("sum__num__yhat_upper", row)
self.assertIn("sum__num__yhat_lower", row)
self.assertEqual(result["rowcount"], 47)
self.assertEqual(result["rowcount"], 103)
@pytest.mark.usefixtures("load_birth_names_dashboard_with_slices")
def test_chart_data_invalid_post_processing(self):