def test_add_source_column(self):
result = render(
pd.DataFrame({"A": [1, 2]}),
params={
"tabs": [
TabOutput(
"tab-2",
"Tab 2",
{"A": RenderColumn("A", "number", "{}")},
pd.DataFrame({"A": [3, 4]}),
)
],
"add_source_column":
True,
"source_column_name":
"S",
},
tab_name="Tab 1",
input_columns={"A": RenderColumn("A", "number", "{}")},
)
expected = pd.DataFrame({
# Source column comes _first_
"S": ["Tab 1", "Tab 1", "Tab 2", "Tab 2"],
"A": [1, 2, 3, 4],
})
# Source column should be categorical: no need to load it with useless
# copied bytes.
expected["S"] = expected["S"].astype("category")
assert_frame_equal(result, expected)
def test_add_source_column(self):
result = render(
pd.DataFrame({'A': [1, 2]}),
params={
'tabs': [
TabOutput('tab-2', 'Tab 2', {
'A': RenderColumn('A', 'number'),
}, pd.DataFrame({'A': [3, 4]})),
],
'add_source_column': True,
'source_column_name': 'S',
},
tab_name='Tab 1',
input_columns={
'A': RenderColumn('A', 'number'),
}
)
expected = pd.DataFrame({
# Source column comes _first_
'S': ['Tab 1', 'Tab 1', 'Tab 2', 'Tab 2'],
'A': [1, 2, 3, 4],
})
# Source column should be categorical: no need to load it with useless
# copied bytes.
expected['S'] = expected['S'].astype('category')
assert_frame_equal(result, expected)
def test_allow_different_columns(self):
result = render(
pd.DataFrame({"A": [1, 2]}),
params={
"tabs": [
TabOutput(
"tab-2",
"Tab 2",
{"B": RenderColumn("B", "number", "{}")},
pd.DataFrame({"B": [3, 4]}),
)
],
"add_source_column":
False,
"source_column_name":
"",
},
tab_name="Tab 1",
input_columns={"A": RenderColumn("A", "number", "{}")},
)
# This tests the ordering of columns, too
assert_frame_equal(
result,
pd.DataFrame({
"A": [1, 2, np.nan, np.nan],
"B": [np.nan, np.nan, 3, 4]
}),
)
def test_error_different_types(self):
result = render(
pd.DataFrame({"A": ["x", "y"]}),
params={
"tabs": [
TabOutput(
"tab-2",
"Tab 2",
{"A": RenderColumn("A", "number", "{}")},
pd.DataFrame({"A": [3, 4]}),
)
],
"add_source_column":
False,
"source_column_name":
"",
},
tab_name="Tab 1",
input_columns={"A": RenderColumn("A", "text", None)},
)
self.assertEqual(
result,
('Cannot concatenate column "A" of type "number" in "Tab 2" to '
'column "A" of type "text" in "Tab 1". Please convert one or the '
"other so they are the same type."),
)
def test_allow_different_columns(self):
result = render(pd.DataFrame({'A': [1, 2]}),
params={
'tabs': [
TabOutput(
'tab-2', 'Tab 2', {
'B': RenderColumn('B', 'number', '{}'),
}, pd.DataFrame({'B': [3, 4]})),
],
'add_source_column':
False,
'source_column_name':
'',
},
tab_name='Tab 1',
input_columns={
'A': RenderColumn('A', 'number', '{}'),
})
# This tests the ordering of columns, too
assert_frame_equal(
result,
pd.DataFrame({
'A': [1, 2, np.nan, np.nan],
'B': [np.nan, np.nan, 3, 4],
}))
def test_happy_path(self):
result = render(
pd.DataFrame({'A': [1, 2]}),
params={
'tabs': [
TabOutput('tab-2', 'Tab 2', {
'A': RenderColumn('A', 'number'),
}, pd.DataFrame({'A': [3, 4]})),
],
'add_source_column': False,
'source_column_name': '',
},
tab_name='Tab 1',
input_columns={
'A': RenderColumn('A', 'number'),
}
)
assert_frame_equal(result, pd.DataFrame({'A': [1, 2, 3, 4]}))
def test_coerce_categories_and_str(self):
result = render(
pd.DataFrame({'A': ['a', 'b']}, dtype='category'), # cat
params={
'tabs': [
TabOutput('tab-2', 'Tab 2', {
'A': RenderColumn('A', 'text'),
}, pd.DataFrame({'A': ['c', 'd']})), # str
],
'add_source_column': False,
'source_column_name': '',
},
tab_name='Tab 1',
input_columns={
'A': RenderColumn('A', 'text'),
}
)
assert_frame_equal(result, pd.DataFrame({
'A': ['a', 'b', 'c', 'd'], # str
}))
def test_happy_path(self):
result = render(
pd.DataFrame({"A": [1, 2]}),
params={
"tabs": [
TabOutput(
"tab-2",
"Tab 2",
{"A": RenderColumn("A", "number", "{}")},
pd.DataFrame({"A": [3, 4]}),
)
],
"add_source_column":
False,
"source_column_name":
"",
},
tab_name="Tab 1",
input_columns={"A": RenderColumn("A", "number", "{}")},
)
assert_frame_equal(result, pd.DataFrame({"A": [1, 2, 3, 4]}))
def test_error_different_types(self):
result = render(
pd.DataFrame({'A': ['x', 'y']}),
params={
'tabs': [
TabOutput('tab-2', 'Tab 2', {
'A': RenderColumn('A', 'number'),
}, pd.DataFrame({'A': [3, 4]})),
],
'add_source_column': False,
'source_column_name': '',
},
tab_name='Tab 1',
input_columns={
'A': RenderColumn('A', 'text'),
}
)
self.assertEqual(result, (
'Cannot concatenate column "A" of type "number" in "Tab 2" to '
'column "A" of type "text" in "Tab 1". Please convert one or the '
'other so they are the same type.'
))
def test_coerce_categories_and_str(self):
result = render(
pd.DataFrame({"A": ["a", "b"]}, dtype="category"), # cat
params={
"tabs": [
TabOutput(
"tab-2",
"Tab 2",
{"A": RenderColumn("A", "text", None)},
pd.DataFrame({"A": ["c", "d"]}),
) # str
],
"add_source_column":
False,
"source_column_name":
"",
},
tab_name="Tab 1",
input_columns={"A": RenderColumn("A", "text", None)},
)
assert_frame_equal(result, pd.DataFrame({"A": ["a", "b", "c",
"d"]})) # str
