/pandas/tests/test_ndframe.py
Python | 30 lines | 20 code | 9 blank | 1 comment | 1 complexity | 3168f1a60e8988a63989ab54f68857fe MD5 | raw file
Possible License(s): BSD-3-Clause
- import unittest
- import numpy as np
- from pandas.core.generic import NDFrame
- import pandas.util.testing as t
- class TestNDFrame(unittest.TestCase):
- def setUp(self):
- tdf = t.makeTimeDataFrame()
- self.ndf = NDFrame(tdf._data)
- def test_constructor(self):
- # with cast
- ndf = NDFrame(self.ndf._data, dtype=int)
- self.assert_(ndf.values.dtype == np.int_)
- def test_ndim(self):
- self.assertEquals(self.ndf.ndim, 2)
- def test_astype(self):
- casted = self.ndf.astype(int)
- self.assert_(casted.values.dtype == np.int_)
- if __name__ == '__main__':
- import nose
- nose.runmodule(argv=[__file__,'-vvs','-x','--pdb', '--pdb-failure'],
- exit=False)