def test_row(self):
df = simple_index.encode_trace(general_example())
row = df[(df.event_nr == 1) & (df.case_id == 3)].iloc[0]
self.assertEqual(1.0, row.prefix_1)
self.assertEqual(0.0, row.elapsed_time)
self.assertEqual(1368780.0, row.remaining_time)
def test_shape(self):
df = simple_index.encode_trace(general_example())
self.assertIn("case_id", df.columns.values)
self.assertIn("event_nr", df.columns.values)
self.assertIn("remaining_time", df.columns.values)
self.assertIn("elapsed_time", df.columns.values)
self.assertIn("prefix_1", df.columns.values)
self.assertEqual((6, 5), df.shape)
def test_shape(self):
df = boolean.encode_trace(general_example())
names = ['register request', 'examine casually', 'check ticket', 'decide',
'reinitiate request', 'examine thoroughly', 'pay compensation',
'reject request', 'case_id', 'event_nr', 'remaining_time',
'elapsed_time']
self.assertListEqual(names, df.columns.values.tolist())
self.assertEqual((42, 12), df.shape)
def test_row(self):
df = frequency.encode_trace(general_example())
row = df[(df.event_nr == 2) & (df.case_id == 2)].iloc[0]
self.assertEqual(1.0, row['register request'])
self.assertEqual(0.0, row['examine casually'])
self.assertEqual(1.0, row['check ticket'])
self.assertEqual(0.0, row['decide'])
self.assertEqual(2400.0, row.elapsed_time)
self.assertEqual(777180.0, row.remaining_time)
def test_encodes_next_activity_prefix(self):
"""Encodes for next activity with prefix length"""
df = simple_index.encode_trace(general_example(),
prefix_length=6,
next_activity=True)
self.assertEqual((6, 8), df.shape)
self.assertIn("prefix_1", df.columns.values)
self.assertIn("prefix_2", df.columns.values)
self.assertIn("prefix_3", df.columns.values)
row = df[df.case_id == 3].iloc[0]
self.assertListEqual([3, 6, 1, 2, 3, 4, 5, 6], row.values.tolist())
def test_prefix_length(self):
df = simple_index.encode_trace(general_example(), prefix_length=3)
self.assertIn("prefix_1", df.columns.values)
self.assertIn("prefix_2", df.columns.values)
self.assertIn("prefix_3", df.columns.values)
self.assertEqual((6, 7), df.shape)
row = df[(df.event_nr == 3) & (df.case_id == 3)].iloc[0]
self.assertEqual(1, row.prefix_1)
self.assertEqual(2, row.prefix_2)
self.assertEqual(3, row.prefix_3)
self.assertEqual(1361460.0, row.remaining_time)
self.assertEqual(7320.0, row.elapsed_time)
def test_encodes_next_activity(self):
"""Encodes for next activity"""
df = simple_index.encode_trace(general_example(), next_activity=True)
self.assertEqual((6, 3), df.shape)
self.assertNotIn("remaining_time", df.columns.values)
self.assertNotIn("elapsed_time", df.columns.values)
self.assertIn("label", df.columns.values)
self.assertNotIn("prefix_1", df.columns.values)
row = df[df.case_id == 3].iloc[0]
self.assertEqual(1, row.label)
self.assertEqual(1, row.event_nr)