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])
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])
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)
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])
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)
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])