Ejemplos de _ensure_2d en Python

Ejemplo n.º 1

Archivo: test_tools.py Proyecto: locolucco209/MongoScraper

 def test_enfore_numpy(self):
     results = tools._ensure_2d(self.df, True)
     assert_array_equal(results[0], self.ndarray)
     assert_array_equal(results[1], self.df.columns)
     results = tools._ensure_2d(self.series, True)
     assert_array_equal(results[0], self.ndarray[:, [0]])
     assert_array_equal(results[1], self.df.columns[0])

Ejemplo n.º 2

Archivo: test_tools.py Proyecto: bfcondon/statsmodels

 def test_enfore_numpy(self):
     results = tools._ensure_2d(self.df, True)
     assert_array_equal(results[0], self.ndarray)
     assert_array_equal(results[1], self.df.columns)
     results = tools._ensure_2d(self.series, True)
     assert_array_equal(results[0], self.ndarray[:,[0]])
     assert_array_equal(results[1], self.df.columns[0])

Ejemplo n.º 3

Archivo: test_tools.py Proyecto: locolucco209/MongoScraper

    def test_numpy(self):
        results = tools._ensure_2d(self.ndarray)
        assert_array_equal(results[0], self.ndarray)
        assert_equal(results[1], None)

        results = tools._ensure_2d(self.ndarray[:, 0])
        assert_array_equal(results[0], self.ndarray[:, [0]])
        assert_equal(results[1], None)

Ejemplo n.º 4

Archivo: test_tools.py Proyecto: locolucco209/MongoScraper

    def test_pandas(self):
        results = tools._ensure_2d(self.df, False)
        assert_frame_equal(results[0], self.df)
        assert_array_equal(results[1], self.df.columns)

        results = tools._ensure_2d(self.series, False)
        assert_frame_equal(results[0], self.df.iloc[:, [0]])
        assert_equal(results[1], self.df.columns[0])

Ejemplo n.º 5

Archivo: test_tools.py Proyecto: bfcondon/statsmodels

    def test_numpy(self):
        results = tools._ensure_2d(self.ndarray)
        assert_array_equal(results[0], self.ndarray)
        assert_equal(results[1], None)

        results = tools._ensure_2d(self.ndarray[:,0])
        assert_array_equal(results[0], self.ndarray[:,[0]])
        assert_equal(results[1], None)

Ejemplo n.º 6

Archivo: test_tools.py Proyecto: bfcondon/statsmodels

    def test_pandas(self):
        results = tools._ensure_2d(self.df, False)
        assert_frame_equal(results[0], self.df)
        assert_array_equal(results[1], self.df.columns)

        results = tools._ensure_2d(self.series, False)
        assert_frame_equal(results[0], self.df.iloc[:,[0]])
        assert_equal(results[1], self.df.columns[0])