def test_render_range_non_numeric_ranges(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(select_range=True, column_numbers="2-3,giraffe"))
self.assertEqual(
result,
('Column numbers must look like "1-2", "5" or "1-2, 5"; ' 'got "giraffe"'),
)
def test_render_range_clamp_range(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(select_range=True, column_numbers="-1,2,6"))
self.assertEqual(
result, 'Column numbers must look like "1-2", "5" or "1-2, 5"; got "-1"'
)
def test_render_range_overlapping_ranges(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(select_range=True, column_numbers="2-3,2"))
self.assertEqual(result, "There are overlapping numbers in input range")
def test_render_range_hyphen_separated(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(select_range=True, column_numbers="2-3", keep=False))
assert_frame_equal(result, pd.DataFrame({"A": [1, 2]}))
def test_render_range_comma_separated(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(select_range=True, column_numbers="1,3"))
assert_frame_equal(result, pd.DataFrame({"A": [1, 2], "C": [3, 4]}))
def test_render_range_ignore_empty_range(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table.copy(), P(select_range=True, column_numbers=""))
self.assertEqual(
result, 'Column numbers must look like "1-2", "5" or "1-2, 5"; got ""'
)
def test_render_drop_columns(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(["B", "C"], keep=False))
assert_frame_equal(result, pd.DataFrame({"A": [1, 2]}))
def test_render_maintain_input_column_order(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(["B"], keep=False))
assert_frame_equal(result, pd.DataFrame({"A": [1, 2], "C": [3, 4]}))
def test_render_single_column(self):
table = pd.DataFrame({"A": [1, 2], "B": [2, 3], "C": [3, 4]})
result = render(table, P(["A"]))
assert_frame_equal(result, pd.DataFrame({"A": [1, 2]}))
def test_render_range_overlapping_ranges(self):
table = pd.DataFrame({'A': [1, 2], 'B': [2, 3], 'C': [3, 4]})
result = render(table, P(select_range=True, column_numbers='2-3,2'))
self.assertEqual(result,
'There are overlapping numbers in input range')
def test_render_range_comma_separated(self):
table = pd.DataFrame({'A': [1, 2], 'B': [2, 3], 'C': [3, 4]})
result = render(table, P(select_range=True, column_numbers='1,3'))
assert_frame_equal(result, pd.DataFrame({'A': [1, 2], 'C': [3, 4]}))
def test_render_drop_columns(self):
table = pd.DataFrame({'A': [1, 2], 'B': [2, 3], 'C': [3, 4]})
result = render(table, P(['B', 'C'], keep=False))
assert_frame_equal(result, pd.DataFrame({'A': [1, 2]}))
def test_render_single_column(self):
table = pd.DataFrame({'A': [1, 2], 'B': [2, 3], 'C': [3, 4]})
result = render(table, P(['A']))
assert_frame_equal(result, pd.DataFrame({'A': [1, 2]}))
