#NEWLINE# def test_event_dag_subset( self ) : #NEWLINE# #INDENT# mockInputEvent = MockInputEvent ( ) #NEWLINE# mockInputEvent2 = MockInputEvent ( ) #NEWLINE# events = [ MockInternalEvent ( 'a' ) , mockInputEvent , mockInputEvent2 ] #NEWLINE# event_dag = EventDag ( mockInputEvent ) #NEWLINE# self . assertEqual ( event_dag . events , event_dag . input_subset ( events ) . events ) #NEWLINE# self . assertEqual ( event_dag . events [ 0 : 1 ] , event_dag . input_subset ( [ ] ) . events ) #NEWLINE# sub_graph = event_dag . input_subset ( [ mockInputEvent ] ) #NEWLINE# self . assertEqual ( event_dag . events [ 0 : 2 ] , sub_graph . events )