linpershey/process_mining

1import pm4py2import os3from pm4py.util import constants, pandas_utils4import importlib.util5import unittest6 7 8class SimplifiedInterface2Test(unittest.TestCase):9    def test_import_ocel_sqlite_1(self):10        ocel = pm4py.read_ocel("input_data/ocel/newocel.sqlite")11 12    def test_import_ocel_sqlite_2(self):13        ocel = pm4py.read_ocel_sqlite("input_data/ocel/newocel.sqlite")14 15    def test_export_ocel_sqlite(self):16        ocel = pm4py.read_ocel("input_data/ocel/newocel.jsonocel")17        pm4py.write_ocel(ocel, "test_output_data/newocel2.sqlite")18        os.remove("test_output_data/newocel2.sqlite")19 20    def test_reduce_invisibles(self):21        net, im, fm = pm4py.read_pnml("input_data/running-example.pnml")22        pm4py.reduce_petri_net_invisibles(net)23 24    def test_reduce_implicit_places(self):25        net, im, fm = pm4py.read_pnml("input_data/running-example.pnml")26        pm4py.reduce_petri_net_implicit_places(net, im, fm)27 28    def test_language_df(self):29        for legacy_obj in [True, False]:30            log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=legacy_obj)31            pm4py.get_stochastic_language(log)32 33    def test_language_log(self):34        log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=True)35        pm4py.get_stochastic_language(log)36 37    def test_language_model(self):38        net, im, fm = pm4py.read_pnml("input_data/running-example.pnml")39        pm4py.get_stochastic_language(net, im, fm)40 41    def test_conversion_df_to_nx(self):42        for legacy_obj in [True, False]:43            log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=legacy_obj)44            pm4py.convert_log_to_networkx(log)45 46    def test_conversion_log_to_nx(self):47        log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=True)48        pm4py.convert_log_to_networkx(log)49 50    def test_conversion_ocel_to_nx(self):51        ocel = pm4py.read_ocel("input_data/ocel/example_log.jsonocel")52        pm4py.convert_ocel_to_networkx(ocel)53 54    def test_conversion_df_to_ocel(self):55        for legacy_obj in [True, False]:56            log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=legacy_obj)57            pm4py.convert_log_to_ocel(log)58 59    def test_conversion_log_to_ocel(self):60        log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=True)61        pm4py.convert_log_to_ocel(log)62 63    def test_conversion_ocelcsv_to_ocel(self):64        dataframe = pandas_utils.read_csv("input_data/ocel/example_log.csv")65        pm4py.convert_log_to_ocel(dataframe, activity_column="ocel:activity", timestamp_column="ocel:timestamp")66 67    def test_conversion_petri_to_nx(self):68        net, im, fm = pm4py.read_pnml("input_data/running-example.pnml")69        nx_digraph = pm4py.convert_petri_net_to_networkx(net, im, fm)70 71    def test_stochastic_language(self):72        if importlib.util.find_spec("pyemd"):73            log1 = pm4py.read_xes("input_data/running-example.xes")74            log2 = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=True)75            lang1 = pm4py.get_stochastic_language(log1)76            lang2 = pm4py.get_stochastic_language(log2)77            pm4py.compute_emd(lang1, lang2)78 79    def test_hybrid_ilp_miner(self):80        for legacy_obj in [True, False]:81            log = pm4py.read_xes("input_data/running-example.xes", return_legacy_log_object=legacy_obj)82            pm4py.discover_petri_net_ilp(log)83 84 85if __name__ == "__main__":86    unittest.main()87