def test_column_wrong_type(self):
# TODO let the module-spec YAML specify type so we don't need to test
# this case.
result = render(pd.DataFrame({'A': [1, 2]}),
P(colname='A', factor=3.0),
input_columns={'A': Column('A', 'text')})
self.assertEqual(result, 'Please select a Number column')
def test_happy_path(self):
result = render(
pd.DataFrame({"A": [1, 2]}),
P(colname="A", factor=3.0),
input_columns={"A": Column("A", "number")},
)
expected = pd.DataFrame({"A": [3.0, 6.0]})
assert_frame_equal(result, expected)
def test_column_wrong_type(self):
# TODO let the module-spec YAML specify type so we don't need to test
# this case.
result = render(
pd.DataFrame({"A": [1, 2]}),
P(colname="A", factor=3.0),
input_columns={"A": Column("A", "text")},
)
self.assertEqual(result, "Please select a Number column")
def test_happy_path(self):
result = render(pd.DataFrame({'A': [1, 2]}),
P(colname='A', factor=3.0),
input_columns={'A': Column('A', 'number')})
expected = pd.DataFrame({'A': [3.0, 6.0]})
assert_frame_equal(result, expected)
def test_no_column_no_op(self):
result = render(pd.DataFrame({'A': [1, 2]}), P(), input_columns={})
expected = pd.DataFrame({'A': [1, 2]})
assert_frame_equal(result, expected)