Codebase list pyexcel-ods / 07ee703 tests / test_filter.py
07ee703

Tree @07ee703 (Download .tar.gz)

test_filter.py @07ee703 — raw · history · blame 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import os

from pyexcel_io import get_data, save_data

from nose.tools import eq_


class TestFilter:
    def setUp(self):
        self.test_file = "test_filter.ods"
        sample = [
            [1, 21, 31],
            [2, 22, 32],
            [3, 23, 33],
            [4, 24, 34],
            [5, 25, 35],
            [6, 26, 36],
        ]
        save_data(self.test_file, sample)
        self.sheet_name = "pyexcel_sheet1"

    def test_filter_row(self):
        filtered_data = get_data(
            self.test_file, start_row=3, library="pyexcel-ods"
        )
        expected = [[4, 24, 34], [5, 25, 35], [6, 26, 36]]
        eq_(filtered_data[self.sheet_name], expected)

    def test_filter_row_2(self):
        filtered_data = get_data(
            self.test_file, start_row=3, row_limit=1, library="pyexcel-ods"
        )
        expected = [[4, 24, 34]]
        eq_(filtered_data[self.sheet_name], expected)

    def test_filter_column(self):
        filtered_data = get_data(
            self.test_file, start_column=1, library="pyexcel-ods"
        )
        expected = [[21, 31], [22, 32], [23, 33], [24, 34], [25, 35], [26, 36]]
        eq_(filtered_data[self.sheet_name], expected)

    def test_filter_column_2(self):
        filtered_data = get_data(
            self.test_file,
            start_column=1,
            column_limit=1,
            library="pyexcel-ods",
        )
        expected = [[21], [22], [23], [24], [25], [26]]
        eq_(filtered_data[self.sheet_name], expected)

    def test_filter_both_ways(self):
        filtered_data = get_data(
            self.test_file, start_column=1, start_row=3, library="pyexcel-ods"
        )
        expected = [[24, 34], [25, 35], [26, 36]]
        eq_(filtered_data[self.sheet_name], expected)

    def test_filter_both_ways_2(self):
        filtered_data = get_data(
            self.test_file,
            start_column=1,
            column_limit=1,
            start_row=3,
            row_limit=1,
            library="pyexcel-ods",
        )
        expected = [[24]]
        eq_(filtered_data[self.sheet_name], expected)

    def tearDown(self):
        os.unlink(self.test_file)