def test_transform_with_other_types(self):
dimensions_as_dictionary = {
"district": {"6", "5", "4"},
"year": {"2020"},
"supervision_type": {"probation", "parole"},
"month": {"11", "12"},
}
dimension_manifest = optimized_metric_big_query_view_exporter.transform_manifest_to_order_enforced_form(
dimensions_as_dictionary)
self.assertEqual(_DIMENSION_MANIFEST, dimension_manifest)
def test_transform_with_other_types(self):
dimensions_as_dictionary = {
'district': {'6', '5', '4'},
'year': {'2020'},
'supervision_type': {'probation', 'parole'},
'month': {'11', '12'},
}
dimension_manifest = optimized_metric_big_query_view_exporter.transform_manifest_to_order_enforced_form(
dimensions_as_dictionary)
self.assertEqual(_DIMENSION_MANIFEST, dimension_manifest)
def test_transform_with_nones(self):
dimensions_as_dictionary = {
"district": {"all", "6", "5", "4"},
"year": {"2020"},
"supervision_type": {"probation", "parole", "none"},
"month": {"11", "12"},
}
expected = [
("district", ["4", "5", "6", "all"]),
("month", ["11", "12"]),
("supervision_type", ["none", "parole", "probation"]),
("year", ["2020"]),
]
dimension_manifest = optimized_metric_big_query_view_exporter.transform_manifest_to_order_enforced_form(
dimensions_as_dictionary)
self.assertEqual(expected, dimension_manifest)
def test_transform_with_nones(self):
dimensions_as_dictionary = {
'district': {'all', '6', '5', '4'},
'year': {'2020'},
'supervision_type': {'probation', 'parole', 'none'},
'month': {'11', '12'},
}
expected = [
('district', ['4', '5', '6', 'all']),
('month', ['11', '12']),
('supervision_type', ['none', 'parole', 'probation']),
('year', ['2020']),
]
dimension_manifest = optimized_metric_big_query_view_exporter.transform_manifest_to_order_enforced_form(
dimensions_as_dictionary)
self.assertEqual(expected, dimension_manifest)
def test_transform_empty(self):
dimension_manifest = optimized_metric_big_query_view_exporter.transform_manifest_to_order_enforced_form(
{})
self.assertEqual([], dimension_manifest)
