def test_event_activity_load_save(self): # Event_Activity object based on Test_Event_Set.test_Event_Activity num_events = 3 event_indexes = array([0, 2]) event_activities = array([[10, 20], [30, 40]]) # Create a dummy event activity object ea1 = Event_Activity(num_events) ea1.set_event_activity(event_activities, event_indexes) # Save it to the temporary dir ea1.save(self.data_dir) # Load the second event activity object from file ea2 = Event_Activity.load(num_events, self.data_dir) assert allclose(ea1.event_activity, ea2.event_activity)
def test_event_activity_load_save(self): # Event_Activity object based on Test_Event_Set.test_Event_Activity num_events = 3 event_indexes = array([0, 2]) event_activities = array([[10, 20], [30, 40]]) # Create a dummy event activity object ea1 = Event_Activity(num_events) ea1.set_event_activity(event_activities, event_indexes) # Save it to the temporary dir ea1.save(self.data_dir) # Load the second event activity object from file ea2 = Event_Activity.load(num_events, self.data_dir) assert allclose(ea1.event_activity, ea2.event_activity)