def sf(cls) -> None: f1 = sf.FrameGO(index=cls._index) for col in range(100): s = sf.Series.from_element(col * .1, index=cls._index[col:col + 20]) f1[col] = s assert f1.sum().sum() == 9900.0
def get_sample_frame_mixed_string_index( size: int = 10000, columns: int = 100) -> tp.Tuple[pd.DataFrame, sf.FrameGO, np.ndarray]: '''Get frames with mixed types. ''' # produces 14950 strings source_ids = list(''.join(x) for x in it.combinations(string.ascii_lowercase, 4)) assert size <= len(source_ids) index = source_ids[:size] cols = source_ids[:columns] dtypes = (float, int, object, bool) sff = sf.FrameGO(index=index) for idx, col in enumerate(cols): s = sf.Series(_typed_array(dtypes[idx % 4], size=size, shift=idx), index=index) sff[col] = s npf = sff.values pdf = pd.DataFrame(index=index) for idx, col in enumerate(cols): s = pd.Series(_typed_array(dtypes[idx % 4], size=size, shift=idx), index=index) pdf[col] = s return pdf, sff, npf
def test_frame_iter_tuple_d(self) -> None: f = sf.FrameGO(index=IndexDate.from_date_range('2020-01-01', '2020-01-03')) post = list(f.iter_tuple(constructor=tuple, axis=0)) self.assertEqual(post, []) post = list(f.iter_tuple(axis=1)) self.assertEqual([len(x) for x in post], [0, 0, 0])
def sf(cls) -> None: index = sf.IndexHierarchy.from_product(list(string.ascii_lowercase), list(string.ascii_lowercase)) f1 = sf.FrameGO(index=index) for col in range(100): s = sf.Series.from_element(col * .1, index=index[col:col + 6]) #pylint: disable=E1136 f1[col] = s assert f1.sum().sum() == 2970.0
def sf(cls): index = sf.IndexHierarchy.from_product(list(string.ascii_lowercase), list(string.ascii_lowercase)) f1 = sf.FrameGO(index=index) for col in range(100): s = sf.Series(col * .1, index=index[col:col + 6]) f1[col] = s assert f1.sum().sum() == 2970.0
def test_frame_iter_array_g(self) -> None: f = sf.FrameGO(index=IndexDate.from_date_range('2020-01-01', '2020-01-03')) post = list(f.iter_array(axis=0)) self.assertEqual(post, []) post = list(f.iter_array(axis=1)) self.assertEqual([x.tolist() for x in post], [[], [], []])
def _get_frame() -> sf.Frame: f = sf.FrameGO(index=np.arange(SHAPE[0])) # >>> columns = tuple('abcdef') for i in range(SHAPE[1]): f[i] = np.arange(SHAPE[0]) return f