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))