fix test failure.

Author: JoyMonteiro

tests/test_components.py

@@ -18,6 +18,7 @@
     TendencyStepper,
     TimeDifferencingWrapper,
     UpdateFrequencyWrapper,
+    set_backend,
 )
 from sympl._core.tracers import reset_packers, reset_tracers
 
@@ -43,6 +44,8 @@
 )
 from climt._core.unyt_backend import UnytBackend, UnytStateContainer
 
+set_backend(UnytBackend())
+
 os.environ["NUMBA_DISABLE_JIT"] = "1"
 
 vertical_dimension_names = ["interface_levels", "mid_levels", "full_levels"]

tests/test_conservation.py

@@ -1,11 +1,17 @@
-from datetime import timedelta
-
 import numpy as np
 import unyt
-from sympl import AdamsBashforth, TendencyComponent, TendencyStepper, get_constant
+from sympl import (
+    AdamsBashforth,
+    TendencyComponent,
+    TendencyStepper,
+    get_constant,
+    set_backend,
+)
 
 import climt
-from climt import UnytTimeDelta
+from climt import UnytBackend, UnytTimeDelta
+
+set_backend(UnytBackend())
 
 
 class ConservationTestBase(object):