def test_get_schema_for_selected_columns_change_order(self):
schema = [('user_id', int32), ('vertex_type', str),
('movie_id', int32), ('rating', int32), ('splits', str)]
selected_schema = FrameSchema.get_schema_for_columns(
schema, ['splits', 'user_id', 'rating'])
self.assertEqual(selected_schema, [('splits', str), ('user_id', int32),
('rating', int32)])
def inspect(self, format_settings=None):
catalog = server.get(self.uri).json()
schema = FrameSchema.from_strings_to_types(catalog["metadata"]['columns'])
data = catalog["data"]
self.rows = map(lambda row: Row(schema, row), data)
default_format_setting = ui.InspectSettings(wrap=len(schema), truncate=10, width=200)
format_settings = default_format_setting if format_settings is None else format_settings
return RowsInspection(data, schema, 0, format_settings)
def inspect(self, format_settings=None):
catalog = server.get(self.uri).json()
schema = FrameSchema.from_strings_to_types(
catalog["metadata"]['columns'])
data = catalog["data"]
self.rows = map(lambda row: Row(schema, row), data)
default_format_setting = ui.InspectSettings(wrap=len(schema),
truncate=10,
width=200)
format_settings = default_format_setting if format_settings is None else format_settings
return RowsInspection(data, schema, 0, format_settings)
def test_get_indices_for_selected_columns(self):
schema = [('user_id', int32), ('vertex_type', str), ('movie_id', int32), ('rating', int32), ('splits', str)]
indices = FrameSchema.get_indices_for_selected_columns(schema, ['user_id', 'splits'])
self.assertEqual(indices, [0, 4])
def test_get_schema_for_selected_columns_change_order(self):
schema = [('user_id', int32), ('vertex_type', str), ('movie_id', int32), ('rating', int32), ('splits', str)]
selected_schema = FrameSchema.get_schema_for_columns(schema, ['splits', 'user_id', 'rating'])
self.assertEqual(selected_schema, [('splits', str), ('user_id', int32), ('rating', int32)])
def test_get_indices_for_selected_columns(self):
schema = [('user_id', int32), ('vertex_type', str),
('movie_id', int32), ('rating', int32), ('splits', str)]
indices = FrameSchema.get_indices_for_selected_columns(
schema, ['user_id', 'splits'])
self.assertEqual(indices, [0, 4])
