def test_onboarding_data_assistant_plot_returns_proper_dict_repr_of_table_domain_chart(
bobby_onboarding_data_assistant_result: OnboardingDataAssistantResult,
) -> None:
plot_result: PlotResult = bobby_onboarding_data_assistant_result.plot_metrics()
table_domain_chart: dict = plot_result.charts[0].to_dict()
assert find_strings_in_nested_obj(table_domain_chart, ["Table Row Count per Batch"])
def test_onboarding_data_assistant_plot_metrics_exclude_column_names_filters_output(
bobby_onboarding_data_assistant_result: OnboardingDataAssistantResult,
) -> None:
exclude_column_names: List[str] = ["VendorID", "passenger_count"]
plot_result: PlotResult = bobby_onboarding_data_assistant_result.plot_metrics(
exclude_column_names=exclude_column_names
)
column_domain_charts: List[dict] = [p.to_dict() for p in plot_result.charts[2:]]
assert len(column_domain_charts) == 73
assert not find_strings_in_nested_obj(column_domain_charts, exclude_column_names)
def test_onboarding_data_assistant_plot_metrics_include_column_names_filters_output(
bobby_onboarding_data_assistant_result: OnboardingDataAssistantResult,
) -> None:
include_column_names: List[str] = ["passenger_count", "trip_distance"]
plot_result: PlotResult = bobby_onboarding_data_assistant_result.plot_metrics(
include_column_names=include_column_names
)
column_domain_charts: List[dict] = [p.to_dict() for p in plot_result.charts[2:]]
assert len(column_domain_charts) == 11 # Normally 85 without filtering
assert find_strings_in_nested_obj(column_domain_charts, include_column_names)
def test_onboarding_data_assistant_plot_returns_proper_dict_repr_of_column_domain_chart(
bobby_onboarding_data_assistant_result: OnboardingDataAssistantResult,
) -> None:
plot_result: PlotResult = bobby_onboarding_data_assistant_result.plot_metrics()
column_domain_charts: List[dict] = [p.to_dict() for p in plot_result.charts[2:]]
assert len(column_domain_charts) == 85
columns: List[str] = [
"VendorID",
"passenger_count",
"RatecodeID",
"store_and_fwd_flag",
"payment_type",
"extra",
"mta_tax",
"tolls_amount",
"improvement_surcharge",
"congestion_surcharge",
]
assert find_strings_in_nested_obj(column_domain_charts, columns)
