Ejemplo n.º 1
0
 def test_dtype_category(self):
     from build import dtype_category, csv_to_dataframe
     res = csv_to_dataframe(filepath)
     new_res = dtype_category(
         res,
         ["employee_id", "company_id", "dept", "join_date", "quit_date"])
     self.assertTrue(isinstance(new_res, pd.DataFrame))
Ejemplo n.º 2
0
    def test_dtype_category(self):
        from build import dtype_category, csv_to_dataframe, merge_dataframe
        res1 = csv_to_dataframe(filepath1)
        self.assertTrue(isinstance(res1, pd.DataFrame))
        res2 = csv_to_dataframe(filepath2)
        self.assertTrue(isinstance(res2, pd.DataFrame))
        res = merge_dataframe(res1, res2, 'user_id')

        new_res = dtype_category(res, ["user_id", "sex", "country", "date", "source", "device", "browser_language", "ads_channel", "browser"])
        self.assertTrue(isinstance(new_res, pd.DataFrame))
Ejemplo n.º 3
0
    def test_dtype_category(self):
        from build import csv_to_dataframe, merge_dataframe, dtype_category
        res1 = csv_to_dataframe(filepath1)
        self.assertTrue(isinstance(res1, pd.DataFrame))
        res2 = csv_to_dataframe(filepath2)
        self.assertTrue(isinstance(res2, pd.DataFrame))
        res = merge_dataframe(res1, res2, 'user_id')

        column_list = [
            "sex", "country", "source", "device", "browser_language",
            "ads_channel", "browser", "conversion", "test"
        ]
        res_new = dtype_category(res, column_list)
        self.assertTrue(isinstance(res_new, pd.DataFrame))
 def test_dtype_category(self):
     from build import dtype_category, csv_to_dataframe
     res = csv_to_dataframe(filepath)
     new_res = dtype_category(res, ["employee_id", "company_id", "dept", "join_date", "quit_date"])
     self.assertTrue(isinstance(new_res, pd.DataFrame))
Ejemplo n.º 5
0
 def test_dtype_category(self):
     from build import dtype_category, csv_to_dataframe
     res = csv_to_dataframe(filepath)
     new_res = dtype_category(res, ["country", "new_user", "source", "converted"])
     self.assertTrue(isinstance(new_res, pd.DataFrame))
Ejemplo n.º 6
0
 def test_dtype_category(self):
     from build import csv_to_dataframe, dtype_category
     res = csv_to_dataframe(filepath)
     res_new = dtype_category(res, ["company_id", "dept"])
     self.assertTrue(isinstance(res_new, pd.DataFrame))