@@ -574,17 +574,20 @@ class TestTrajectoryCollectionPlotPrediction(unittest.TestCase):
574574 num_species : int
575575 NUM_POINT : int = 50
576576 tc : TrajectoryCollection
577- END_TIME : float = 1.5
577+ END_TIME : float = 0.1
578+ MODEL_NUM : int = 2
579+ MODEL_NAME : str = makeBiomdName (MODEL_NUM )
578580
579581 @classmethod
580582 def setUpClass (cls ) -> None :
581583 #if IGNORE_TESTS:
582584 # return
583- if not os .path .exists (BIOMD153_PATH ):
584- raise unittest .SkipTest (f"{ makeBiomdName (153 )} not found at { BIOMD153_PATH } " )
585+ path = makeBiomdPath (cls .MODEL_NUM )
586+ if not os .path .exists (path ):
587+ raise unittest .SkipTest (f"{ cls .MODEL_NAME } not found at { path } " )
585588 cls .trajectory = Trajectory .makeBiomodel (
586- path = BIOMD153_PATH , end_time = cls .END_TIME , num_point = cls .NUM_POINT )
587- cls .tc = TrajectoryCollection .split (cls .trajectory , [0.5 ])
589+ path = path , end_time = cls .END_TIME , num_point = cls .NUM_POINT )
590+ cls .tc = TrajectoryCollection .split (cls .trajectory , [cls . END_TIME / 2 ])
588591 if IGNORE_TESTS :
589592 print (pd .DataFrame (cls .trajectory .jacobian_median_arr ))
590593 print (pd .DataFrame (cls .tc .trajectories [0 ].jacobian_median_arr ))
@@ -605,8 +608,8 @@ def test_jacobian_collection_sizes(self) -> None:
605608 def test_plot (self ):
606609 if IGNORE_TESTS :
607610 return
608- self .trajectory .plotPrediction (title = "BIOMD153" )
609- self .tc .plotPrediction (title = "BIOMD153" )
611+ self .trajectory .plotPrediction (title = self . MODEL_NAME , ylim = ( 0.0 , 1e-5 ) )
612+ self .tc .plotPrediction (title = self . MODEL_NAME , ylim = ( 0.0 , 1e-5 ) )
610613 if IS_PLOT :
611614 plt .show ()
612615
0 commit comments