linpershey/process_mining

1import unittest2import os3 4from pm4py.objects.log.importer.xes import importer as xes_importer5from pm4py.objects.log.util import dataframe_utils6from pm4py.util import constants, pandas_utils7from pm4py.algo.organizational_mining.roles import algorithm as role_mining8 9 10class RoleDetectionTest(unittest.TestCase):11    def test_role_running_csv(self):12        df = pandas_utils.read_csv(os.path.join("input_data", "running-example.csv"))13        df = dataframe_utils.convert_timestamp_columns_in_df(df, timest_format=constants.DEFAULT_TIMESTAMP_PARSE_FORMAT)14        roles = role_mining.apply(df)15 16    def test_role_running_xes(self):17        log = xes_importer.apply(os.path.join("..", "tests", "input_data", "running-example.xes"))18        roles = role_mining.apply(log)19 20    def test_role_receipt_csv(self):21        df = pandas_utils.read_csv(os.path.join("input_data", "receipt.csv"))22        df = dataframe_utils.convert_timestamp_columns_in_df(df, timest_format=constants.DEFAULT_TIMESTAMP_PARSE_FORMAT)23        roles = role_mining.apply(df)24 25    def test_role_receipt_xes(self):26        log = xes_importer.apply(os.path.join("..", "tests", "input_data", "receipt.xes"))27        roles = role_mining.apply(log)28 29 30if __name__ == "__main__":31    unittest.main()32