Python assert_sp_array_equalの例

コード例 #1

ファイル: test_list.py プロジェクト: c0state/pandas

    def test_append_zero(self):
        arr = self.zero_data
        splist = SparseList(fill_value=0)
        splist.append(arr[:5])
        splist.append(arr[5])
        splist.append(arr[6:])

        sparr = splist.to_array()
        assert_sp_array_equal(sparr, SparseArray(arr, fill_value=0))

コード例 #2

ファイル: test_list.py プロジェクト: zkluo1/pandas

    def test_append_zero(self):
        arr = self.zero_data
        splist = SparseList(fill_value=0)
        splist.append(arr[:5])
        splist.append(arr[5])
        splist.append(arr[6:])

        sparr = splist.to_array()
        assert_sp_array_equal(sparr, SparseArray(arr, fill_value=0))

コード例 #3

ファイル: test_list.py プロジェクト: c0state/pandas

    def test_append_na(self):
        arr = self.na_data
        splist = SparseList()
        splist.append(arr[:5])
        splist.append(arr[5])
        splist.append(arr[6:])

        sparr = splist.to_array()
        assert_sp_array_equal(sparr, SparseArray(arr))

コード例 #4

ファイル: test_list.py プロジェクト: zkluo1/pandas

    def test_append_na(self):
        arr = self.na_data
        splist = SparseList()
        splist.append(arr[:5])
        splist.append(arr[5])
        splist.append(arr[6:])

        sparr = splist.to_array()
        assert_sp_array_equal(sparr, SparseArray(arr))

コード例 #5

ファイル: test_list.py プロジェクト: c0state/pandas

    def test_copy(self):
        arr = self.na_data
        exp_sparr = SparseArray(arr)

        splist = SparseList()
        splist.append(arr[:5])
        splist.append(arr[5])

        cp = splist.copy()
        cp.append(arr[6:])
        self.assertEquals(splist.nchunks, 2)
        assert_sp_array_equal(cp.to_array(), exp_sparr)

コード例 #6

ファイル: test_list.py プロジェクト: zkluo1/pandas

    def test_copy(self):
        arr = self.na_data
        exp_sparr = SparseArray(arr)

        splist = SparseList()
        splist.append(arr[:5])
        splist.append(arr[5])

        cp = splist.copy()
        cp.append(arr[6:])
        self.assertEquals(splist.nchunks, 2)
        assert_sp_array_equal(cp.to_array(), exp_sparr)

コード例 #7

ファイル: test_list.py プロジェクト: c0state/pandas

    def test_consolidate(self):
        arr = self.na_data
        exp_sparr = SparseArray(arr)

        splist = SparseList()
        splist.append(arr[:5])
        splist.append(arr[5])
        splist.append(arr[6:])

        consol = splist.consolidate(inplace=False)
        self.assert_(consol.nchunks == 1)
        self.assert_(splist.nchunks == 3)
        assert_sp_array_equal(consol.to_array(), exp_sparr)

        splist.consolidate()
        self.assert_(splist.nchunks == 1)
        assert_sp_array_equal(splist.to_array(), exp_sparr)

コード例 #8

ファイル: test_list.py プロジェクト: zkluo1/pandas

    def test_consolidate(self):
        arr = self.na_data
        exp_sparr = SparseArray(arr)

        splist = SparseList()
        splist.append(arr[:5])
        splist.append(arr[5])
        splist.append(arr[6:])

        consol = splist.consolidate(inplace=False)
        self.assert_(consol.nchunks == 1)
        self.assert_(splist.nchunks == 3)
        assert_sp_array_equal(consol.to_array(), exp_sparr)

        splist.consolidate()
        self.assert_(splist.nchunks == 1)
        assert_sp_array_equal(splist.to_array(), exp_sparr)

コード例 #9

ファイル: test_list.py プロジェクト: c0state/pandas

def assert_sp_list_equal(left, right):
    assert_sp_array_equal(left.to_array(), right.to_array())

コード例 #10

ファイル: test_list.py プロジェクト: zkluo1/pandas

def assert_sp_list_equal(left, right):
    assert_sp_array_equal(left.to_array(), right.to_array())