@@ -329,23 +329,21 @@ def test_trials_from_block_n_trials(self) -> None:
329329 """
330330 Test to get the number of trials.
331331 """
332- self .assertEqual (self .trial_object .n_trials , len ( self . block . segments ) )
332+ self .assertEqual (self .trial_object .n_trials , 36 )
333333
334334 def test_trials_from_block_n_spiketrains_trial_by_trial (self ) -> None :
335335 """
336336 Test to get the number of `SpikeTrain` objects per trial.
337337 """
338338 self .assertEqual (self .trial_object .n_spiketrains_trial_by_trial ,
339- [len (trial .spiketrains ) for trial in
340- self .block .segments ])
339+ [2 ] * 36 )
341340
342341 def test_trials_from_block_n_analogsignals_trial_by_trial (self ) -> None :
343342 """
344343 Test to get the number of `AnalogSignal` objects per trial.
345344 """
346345 self .assertEqual (self .trial_object .n_analogsignals_trial_by_trial ,
347- [len (trial .analogsignals ) for trial in
348- self .block .segments ])
346+ [2 ] * 36 )
349347
350348 def test_trials_from_block_get_spiketrains_from_trial_as_list (self
351349 ) -> None :
@@ -636,23 +634,21 @@ def test_trials_from_list_n_trials(self) -> None:
636634 """
637635 Test to get the number of trials.
638636 """
639- self .assertEqual (self .trial_object .n_trials , len ( self . trial_list ) )
637+ self .assertEqual (self .trial_object .n_trials , 36 )
640638
641639 def test_trials_from_list_n_spiketrains_trial_by_trial (self ) -> None :
642640 """
643641 Test to get the number of `SpikeTrain` objects per trial.
644642 """
645643 self .assertEqual (self .trial_object .n_spiketrains_trial_by_trial ,
646- [sum (map (lambda x : isinstance (x , SpikeTrain ),
647- trial )) for trial in self .trial_list ])
644+ [self .n_spiketrains ] * 36 )
648645
649646 def test_trials_from_list_n_analogsignals_trial_by_trial (self ) -> None :
650647 """
651648 Test to get the number of `AnalogSignal` objects per trial.
652649 """
653650 self .assertEqual (self .trial_object .n_analogsignals_trial_by_trial ,
654- [sum (map (lambda x : isinstance (x , AnalogSignal ),
655- trial )) for trial in self .trial_list ])
651+ [self .n_analogsignals ] * 36 )
656652
657653 def test_trials_from_list_get_spiketrains_from_trial_as_list (self ) -> None :
658654 """
0 commit comments