fix tests

brainunit/_base_test.py

@@ -13,23 +13,24 @@
 # limitations under the License.
 # ==============================================================================
 
+from __future__ import annotations
 
+import itertools
 import os
+import pickle
+import sys
 import tempfile
-
-os.environ['JAX_TRACEBACK_FILTERING'] = 'off'
-import itertools
 import unittest
 import warnings
 from copy import deepcopy
+from typing import Union
 
 import brainstate as bst
 import jax
 import jax.numpy as jnp
 import numpy as np
 import pytest
 from numpy.testing import assert_equal
-from typing import Union
 
 import brainunit as u
 from brainunit._base import (
@@ -49,7 +50,6 @@
 )
 from brainunit._unit_common import *
 from brainunit._unit_shortcuts import kHz, ms, mV, nS
-import pickle
 
 
 class TestDimension(unittest.TestCase):
@@ -902,6 +902,9 @@ def test_to(self):
         print(x.to(u.uvolt))
 
     def test_quantity_type(self):
+
+        # if sys.version_info >= (3, 11):
+
         def f1(a: u.Quantity[u.ms]) -> u.Quantity[u.mV]:
             return a