Esempio n. 1
0
    def test_get_dtype_kinds_period(self):
        # because we don't have Period dtype (yet),
        # Series results in object dtype
        to_concat = [
            pd.PeriodIndex(['2011-01'], freq='M'),
            pd.PeriodIndex(['2011-01'], freq='M')
        ]
        res = _concat.get_dtype_kinds(to_concat)
        assert res == set(['period[M]'])

        to_concat = [
            pd.Series([pd.Period('2011-01', freq='M')]),
            pd.Series([pd.Period('2011-02', freq='M')])
        ]
        res = _concat.get_dtype_kinds(to_concat)
        assert res == set(['object'])

        to_concat = [
            pd.PeriodIndex(['2011-01'], freq='M'),
            pd.PeriodIndex(['2011-01'], freq='D')
        ]
        res = _concat.get_dtype_kinds(to_concat)
        assert res == set(['period[M]', 'period[D]'])

        to_concat = [
            pd.Series([pd.Period('2011-01', freq='M')]),
            pd.Series([pd.Period('2011-02', freq='D')])
        ]
        res = _concat.get_dtype_kinds(to_concat)
        assert res == set(['object'])
Esempio n. 2
0
    def test_get_dtype_kinds_period(self):
        # because we don't have Period dtype (yet),
        # Series results in object dtype
        to_concat = [pd.PeriodIndex(['2011-01'], freq='M'),
                     pd.PeriodIndex(['2011-01'], freq='M')]
        res = _concat.get_dtype_kinds(to_concat)
        self.assertEqual(res, set(['period[M]']))

        to_concat = [pd.Series([pd.Period('2011-01', freq='M')]),
                     pd.Series([pd.Period('2011-02', freq='M')])]
        res = _concat.get_dtype_kinds(to_concat)
        self.assertEqual(res, set(['object']))

        to_concat = [pd.PeriodIndex(['2011-01'], freq='M'),
                     pd.PeriodIndex(['2011-01'], freq='D')]
        res = _concat.get_dtype_kinds(to_concat)
        self.assertEqual(res, set(['period[M]', 'period[D]']))

        to_concat = [pd.Series([pd.Period('2011-01', freq='M')]),
                     pd.Series([pd.Period('2011-02', freq='D')])]
        res = _concat.get_dtype_kinds(to_concat)
        self.assertEqual(res, set(['object']))
Esempio n. 3
0
def test_get_dtype_kinds_period(to_concat, expected):
    result = _concat.get_dtype_kinds(to_concat)
    assert result == set(expected)
Esempio n. 4
0
def test_get_dtype_kinds(klass, to_concat, expected):
    to_concat_klass = [klass(c) for c in to_concat]
    result = _concat.get_dtype_kinds(to_concat_klass)
    assert result == set(expected)
Esempio n. 5
0
def test_get_dtype_kinds_period(to_concat, expected):
    result = _concat.get_dtype_kinds(to_concat)
    assert result == set(expected)
Esempio n. 6
0
def test_get_dtype_kinds(klass, to_concat, expected):
    to_concat_klass = [klass(c) for c in to_concat]
    result = _concat.get_dtype_kinds(to_concat_klass)
    assert result == set(expected)
Esempio n. 7
0
 def check_concat(self, to_concat, exp):
     for klass in [pd.Index, pd.Series]:
         to_concat_klass = [klass(c) for c in to_concat]
         res = _concat.get_dtype_kinds(to_concat_klass)
         assert res == set(exp)
Esempio n. 8
0
 def check_concat(self, to_concat, exp):
     for klass in [pd.Index, pd.Series]:
         to_concat_klass = [klass(c) for c in to_concat]
         res = _concat.get_dtype_kinds(to_concat_klass)
         self.assertEqual(res, set(exp))