@@ -0,0 +1,941 @@
+from bndl.compute.tests import DatasetTest%0Aimport numpy as np%0A%0A%0Aclass DataFrameTest(DatasetTest):%0A    df_size = 100%0A%0A    @classmethod%0A    def setUpClass(cls):%0A        super().setUpClass()%0A        cls.df = cls.ctx.range(cls.df_size).map(lambda i: dict(num=i, txt=str(i))).as_dataframe()%0A%0A    def test_from_range(self):%0A        df = self.df.collect()%0A        self.assertTrue((df%5B'num'%5D == np.arange(100)).all())%0A        self.assertTrue((df%5B'txt'%5D == list(map(str, range(100)))).all())%0A        self.assertTrue((self.df%5B'num'%5D.collect() == list(range(self.df_size))).all())%0A%0A    def test_assign(self):%0A        nums2 = self.df%5B'num'%5D.map(lambda i: i ** 2)%0A        with_nums2 = self.df.assign('num2', nums2).collect()%0A%0A        self.assertIn('num', with_nums2)%0A        self.assertIn('num2', with_nums2)%0A%0A        self.assertTrue((with_nums2%5B'num'%5D == np.arange(100)).all())%0A        self.assertTrue((with_nums2%5B'num2'%5D == np.arange(100) ** 2).all())%0A

add df test