fix tests

chaoming0625 · Aug 23, 2024 · ab76554 · ab76554
1 parent d456f0e
commit ab76554
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/brainunit/_base_test.py b/brainunit/_base_test.py
@@ -97,7 +97,7 @@ def test_inplace_operations(self):
         inplace_op(volt)
 
   def test_display(self):
-    assert_equal(str(bu.kmeter / bu.meter), 'Unit(1000.0)')
+    assert_equal(str(bu.kmeter / bu.meter), 'Unit(10.0^3)')
 
 
 class TestQuantity(unittest.TestCase):
@@ -243,9 +243,8 @@ def test_display(self):
       assert_equal(display_in_unit(10. * mV, ohm * amp), "0.01 * volt")
       with pytest.raises(bu.UnitMismatchError):
         display_in_unit(10 * nS, ohm)
-    assert_equal(display_in_unit(10.0, Unit(scale=1)), "1. * Unit(10.0)")
-
-    assert_equal(display_in_unit(3 * bu.kmeter / bu.meter), '3. * Unit(1000.0)')
+    assert_equal(display_in_unit(10.0, Unit(scale=1)), "1. * Unit(10.0^1)")
+    assert_equal(display_in_unit(3 * bu.kmeter / bu.meter), '3. * Unit(10.0^3)')
 
   def test_display2(self):