def test_archive_components_npy_contents_b(self) -> None: f1 = ff.parse('s(2,4)|v(int,str,bool,bool)').relabel(index=('a', 'b')) with TemporaryDirectory() as fp: f1.to_npy(fp) with self.assertRaises(UnsupportedOperation): _ = NPY(fp, 'w').contents
def test_archive_components_npy_nbytes_a(self) -> None: f1 = ff.parse('s(2,4)|v(int,str,bool,bool)').relabel(index=('a', 'b')) with TemporaryDirectory() as fp: f1.to_npy(fp) npy = NPY(fp, 'r') self.assertEqual(npy.contents['size'].sum(), npy.nbytes)
def test_archive_components_npz_from_frames_k(self) -> None: f1 = ff.parse('s(2,2)|v(int)').relabel(index=('a', 'b')) f2 = ff.parse('s(2,2)|v(int)').relabel(index=('c', 'd')) with TemporaryDirectory() as fp: with self.assertRaises(RuntimeError): NPY(fp, 'w').from_frames(frames=(f1, f2), axis=3)
def test_archive_components_npz_from_frames_m(self) -> None: f1 = ff.parse('s(2,2)|v(int)').relabel(index=('a', 'b')) f2 = ff.parse('s(2,2)|v(int)').relabel(index=('c', 'd')) with TemporaryDirectory() as fp: with self.assertRaises(UnsupportedOperation): NPY(fp, 'r').from_frames(frames=(f1, f2), axis=3)
def test_archive_components_npz_from_frames_f(self) -> None: f1 = ff.parse('s(2,2)|v(int)') f2 = ff.parse('s(2,2)|v(int)') with TemporaryDirectory() as fp: with self.assertRaises(RuntimeError): NPY(fp, 'w').from_frames(frames=(f1, f2), axis=1)
def test_archive_components_npy_write_arrays_k(self) -> None: a1 = np.arange(12).reshape(3, 4) a2 = np.array(['a', 'b', 'c']) a3 = np.array([True, False, True]) with TemporaryDirectory() as fp: with self.assertRaises(UnsupportedOperation): NPY(fp, 'r').from_arrays(blocks=(a1, a2, a3))
def test_archive_components_npz_from_frames_j(self) -> None: f1 = ff.parse('s(2,2)|v(float)').relabel(columns=('a', 'b')) f2 = ff.parse('s(2,2)|v(float)').relabel(columns=('b', 'c')) with TemporaryDirectory() as fp: NPY(fp, 'w').from_frames(frames=(f1, f2), axis=0, include_index=False) f = Frame.from_npy(fp).fillna(0) self.assertEqual(f.to_pairs(), (('a', ((0, 1930.4), (1, -1760.34), (2, 0.0), (3, 0.0))), ('b', ((0, -610.8), (1, 3243.94), (2, 1930.4), (3, -1760.34))), ('c', ((0, 0.0), (1, 0.0), (2, -610.8), (3, 3243.94)))) )
def test_archive_components_npy_write_arrays_i(self) -> None: a1 = np.arange(12).reshape(3, 4) a2 = np.array(['a', 'b', 'c']) a3 = np.array([True, False, True]) with TemporaryDirectory() as fp: columns=('a', 'b', 'c', 'd', 'e', 'f') with self.assertRaises(RuntimeError): NPY(fp, 'w').from_arrays(blocks=(a1, a2, a3), columns=columns, name='bar')
def test_archive_components_npz_from_frames_d(self) -> None: f1 = ff.parse('s(2,2)|v(int)').relabel(columns=('a', 'b')) f2 = ff.parse('s(2,2)|v(int)').relabel(columns=('c', 'd')) with TemporaryDirectory() as fp: NPY(fp, 'w').from_frames(frames=(f1, f2), axis=1, include_columns=False) f = Frame.from_npy(fp) self.assertEqual(f.to_pairs(), ((0, ((0, -88017), (1, 92867))), (1, ((0, 162197), (1, -41157))), (2, ((0, -88017), (1, 92867))), (3, ((0, 162197), (1, -41157)))) )
def test_archive_components_npy_write_arrays_j(self) -> None: a1 = np.arange(12).reshape(3, 4) a2 = np.array(['a', 'b', 'c']) a3 = np.array([True, False, True]) with TemporaryDirectory() as fp: columns=np.arange(6).astype('datetime64[D]') NPY(fp, 'w').from_arrays(blocks=(a1, a2, a3), columns=columns, name='bar', axis=1) f = Frame.from_npy(fp) dt64 = np.datetime64 self.assertEqual(f.to_pairs(), ((dt64('1970-01-01'), ((0, 0), (1, 4), (2, 8))), (dt64('1970-01-02'), ((0, 1), (1, 5), (2, 9))), (dt64('1970-01-03'), ((0, 2), (1, 6), (2, 10))), (dt64('1970-01-04'), ((0, 3), (1, 7), (2, 11))), (dt64('1970-01-05'), ((0, 'a'), (1, 'b'), (2, 'c'))), (dt64('1970-01-06'), ((0, True), (1, False), (2, True)))) )
def test_archive_components_npy_write_arrays_h(self) -> None: a1 = np.arange(12).reshape(3, 4) a2 = np.array(['a', 'b', 'c']) a3 = np.array([True, False, True]) with TemporaryDirectory() as fp: columns=Index(('a', 'b', 'c', 'd', 'e', 'f'), name='foo') NPY(fp, 'w').from_arrays(blocks=(a1, a2, a3), columns=columns, name='bar', axis=1) f = Frame.from_npy(fp) self.assertEqual(f.to_pairs(), (('a', ((0, 0), (1, 4), (2, 8))), ('b', ((0, 1), (1, 5), (2, 9))), ('c', ((0, 2), (1, 6), (2, 10))), ('d', ((0, 3), (1, 7), (2, 11))), ('e', ((0, 'a'), (1, 'b'), (2, 'c'))), ('f', ((0, True), (1, False), (2, True)))) ) self.assertEqual(f.name, 'bar') self.assertEqual(f.columns.name, 'foo')