def test_odd_row_filter(self):
r = pe.ColumnSeriesReader(self.test_tuple)
f = pe.sheets.filters.OddRowFilter()
r.add_filter(f)
actual = pe.utils.to_dict(r)
result = {"Y": [11, 21, 31, 41, 51]}
assert result == actual
def test_orthogonality2(self):
r = pe.ColumnSeriesReader(self.test_tuple)
r.add_filter(pe.sheets.filters.OddColumnFilter())
r.add_filter(pe.sheets.filters.EvenRowFilter())
actual = pe.utils.to_dict(r)
result = {"X": [2, 4], "Z": [22, 42]}
assert result == actual
def test_row_filter(self):
r = pe.ColumnSeriesReader(self.test_tuple)
r.add_filter(pe.filters.SingleRowFilter(1))
actual = pe.utils.to_dict(r)
result = {
"X": [1, 2, 3, 4, 5],
"Z": [12, 22, 32, 42, 52]
}
assert result == actual
def test_even_row_filter(self):
r = pe.ColumnSeriesReader(self.test_tuple)
r.add_filter(pe.filters.EvenRowFilter())
actual = pe.utils.to_dict(r)
result = {"X": [1, 2, 3, 4, 5], "Z": [12, 22, 32, 42, 52]}
assert result == actual
# test removing the filter, it prints the original one
r.clear_filters()
actual = pe.utils.to_array(r.rows())
assert actual == pe.transpose(self.content[1:])
def test_odd_row_filter(self):
r = pe.ColumnSeriesReader(self.test_tuple)
f = pe.filters.OddRowFilter()
r.add_filter(f)
actual = pe.utils.to_dict(r)
result = {"Y": [11, 21, 31, 41, 51]}
assert result == actual
r.remove_filter(f)
actual = pe.utils.to_array(r.rows())
assert actual == pe.transpose(self.content[1:])
def test_column_filter(self):
r = pe.ColumnSeriesReader(self.test_tuple)
r.add_filter(pe.sheets.filters.SingleColumnFilter(0))
actual = pe.utils.to_dict(r)
result = {
"X": [2, 3, 4, 5],
"Y": [21, 31, 41, 51],
"Z": [22, 32, 42, 52]
}
assert result == actual
assert r.rownames == ["X", "Y", "Z"]
def test_orthogonality(self):
r = pe.ColumnSeriesReader(self.test_tuple)
r.add_filter(pe.filters.EvenRowFilter())
r.add_filter(pe.filters.OddColumnFilter())
actual = pe.to_dict(r)
result = {"X": [2, 4], "Z": [22, 42]}
assert result == actual
# test removing the filter, it prints the original one
r.clear_filters()
actual = pe.utils.to_array(r.rows())
assert actual == pe.transpose(self.content[1:])
def test_series_column_iterator(self):
r = pe.ColumnSeriesReader(self.test_tuple)
sri = pe.sheets.iterators.RowIndexIterator(r)
actual = pe.utils.to_array(sri)
result = [{
'X': [1, 2, 3, 4, 5]
}, {
'Y': [11, 21, 31, 41, 51]
}, {
'Z': [12, 22, 32, 42, 52]
}]
assert actual == result
def test_index_sheet2(self):
s = pe.ColumnSeriesReader(self.testfile, series=0)
result = [{
'1': 4,
'X': 'Y',
'3': 6,
'2': 5
}, {
'1': 7,
'X': 'Z',
'3': 9,
'2': 8
}]
self.assertEqual(result, s.records)
def test_index_sheet2(self):
s = pe.ColumnSeriesReader(self.testfile, series=0)
actual = pe.to_records(s)
result = [{
'1.0': 4.0,
'X': 'Y',
'3.0': 6.0,
'2.0': 5.0
}, {
'1.0': 7.0,
'X': 'Z',
'3.0': 9.0,
'2.0': 8.0
}]
assert actual == result
def test_orthogonality2(self):
r = pe.ColumnSeriesReader(self.test_tuple)
r.add_filter(pe.filters.OddColumnFilter())
r.add_filter(pe.filters.EvenRowFilter())
actual = pe.utils.to_dict(r)
result = {
"X": [2, 4],
"Z": [22, 42]
}
assert result == actual
# test removing the filter, it prints the original one
r.clear_filters()
actual = pe.utils.to_array(r)
result = [{'X': [1, 2, 3, 4, 5]}, {'Y': [11, 21, 31, 41, 51]}, {'Z': [12, 22, 32, 42, 52]}]
assert actual == result
def test_index_sheet2(self):
s = pe.ColumnSeriesReader(self.testfile, series=0)
result = [
{
"1": 4,
"X": "Y",
"3": 6,
"2": 5
},
{
"1": 7,
"X": "Z",
"3": 9,
"2": 8
},
]
eq_(result, list(s.records))
def test_index_sheet3(self):
s = pe.ColumnSeriesReader(self.testfile, series=0)
headers = ["Row 1", "Row 2", "Row 3", "Row 4"]
actual = pe.to_records(s, headers)
print(actual)
result = [{
'Row 4': 6.0,
'Row 2': 4.0,
'Row 1': 'Y',
'Row 3': 5.0
}, {
'Row 4': 9.0,
'Row 2': 7.0,
'Row 1': 'Z',
'Row 3': 8.0
}]
assert actual == result
def test_index_sheet3(self):
s = pe.ColumnSeriesReader(self.testfile, series=0)
headers = ["Row 1", "Row 2", "Row 3", "Row 4"]
actual = s.to_records(headers)
result = [
{
"Row 4": 6.0,
"Row 2": 4.0,
"Row 1": "Y",
"Row 3": 5.0
},
{
"Row 4": 9.0,
"Row 2": 7.0,
"Row 1": "Z",
"Row 3": 8.0
},
]
eq_(result, list(actual))
