small fix for test

esa · Mar 18, 2024 · eebabd6 · eebabd6
1 parent 8031f67
commit eebabd6
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_batched_sgp4.py b/tests/test_batched_sgp4.py
@@ -30,11 +30,13 @@ def test_sgp4_batched(self):
             try:
                 dsgp4.initialize_tle(tle,gravity_constant_name="wgs-72");
                 #we only check those that have not failed to initialize:
-                if tle._error==0:   
-                    tsinces=torch.rand(100)*10
-                    tles_batch+=[tle]*len(tsinces)
-                    tsinces_batch+=[tsinces]
-                    out_non_batched+=[dsgp4.propagate(tle,tsinces)]
+                if tle._error==0:
+                    #batch mode only supports isimp==0
+                    if tle._isimp==0:       
+                        tsinces=torch.rand(100)*10
+                        tles_batch+=[tle]*len(tsinces)
+                        tsinces_batch+=[tsinces]
+                        out_non_batched+=[dsgp4.propagate(tle,tsinces)]
             except Exception as e:
                     self.assertTrue((str(e).split()==error_string.split()) or ((str(e).split()==error_string_isimp.split())))
         tsinces_batch=torch.cat(tsinces_batch)