def test_first_row_is_header_autocast_dtypes(self):
# TODO make fetch_result _not_ a pd.DataFrame
fetch_result = ProcessResult(pd.DataFrame({"A": ["nums", "3"]}))
result = scrapetable.render(pd.DataFrame(),
P(first_row_is_header=True),
fetch_result=fetch_result)
assert_frame_equal(result, pd.DataFrame({"nums": [3]}))
def test_first_row_is_header_zero_rows(self):
# TODO make fetch_result _not_ a pd.DataFrame
fetch_result = ProcessResult(pd.DataFrame({"A": [], "B": []}))
result = scrapetable.render(pd.DataFrame(),
P(first_row_is_header=True),
fetch_result=fetch_result)
assert_frame_equal(result, pd.DataFrame({"A": [], "B": []}))
def test_first_row_is_header_empty_values(self):
# TODO make fetch_result _not_ a pd.DataFrame
fetch_result = ProcessResult(pd.DataFrame({"A": ["", "x"], "B": [2, 3]}))
result = scrapetable.render(
pd.DataFrame(), P(first_row_is_header=True), fetch_result=fetch_result
)
assert_frame_equal(result, pd.DataFrame({"Column 1": ["x"], "2": [3]}))
def test_first_row_is_header(self):
# TODO make fetch_result _not_ a pd.DataFrame, so we don't lose info
# when converting types here
fetch_result = ProcessResult(pd.DataFrame(a_table.copy()))
result = scrapetable.render(pd.DataFrame(),
P(first_row_is_header=True),
fetch_result=fetch_result)
assert_frame_equal(result, pd.DataFrame({"1": [2], "2": [3]}))
