def test_get_item_operator(self):
r = p.SeriesReader(self.testfile, series=4)
value = r[0, 1]
assert value == 2
def test_headers(self):
r = p.SeriesReader(self.testfile, series=4)
actual = r.colnames
eq_(self.content[4], actual)
def test_named_column_at(self):
r = p.SeriesReader(self.testfile, series=4)
result = r.named_column_at("X")
actual = {"X": [1, 1, 1, 1, 1]}
eq_(result, actual["X"])
def test_content_is_read(self):
r = p.SeriesReader(self.testfile, series=4)
actual = list(r.rows())
self.content.pop(4)
eq_(self.content, actual)
def test_headers(self):
r = p.SeriesReader(self.testfile)
actual = r.colnames
assert self.content[0] == actual
def test_content_is_read(self):
r = p.SeriesReader(self.testfile)
actual = list(r.rows())
assert self.content[1:] == actual
def test_set_named_column_at(self):
r = pe.SeriesReader(self.testfile)
r.set_named_column_at("b", [11, 1])
eq_(r.column_at(1), [11, 1])
def test_non_default_sheet_as_single_sheet_reader_series(self):
r = pe.SeriesReader(self.testfile, "Sheet3")
data = pe.utils.to_array(r.rows())
assert data == self.content["Sheet3"][1:]