From 32e2c074e184078eeac6904722604304c44567f4 Mon Sep 17 00:00:00 2001 From: Dhruv Sondhi Date: Fri, 2 Jul 2021 13:10:12 +0530 Subject: [PATCH] Fixing some errors with the tests --- tardis/io/logger/tests/test_logging.py | 3 ++- tardis/simulation/tests/test_simulation.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tardis/io/logger/tests/test_logging.py b/tardis/io/logger/tests/test_logging.py index cc46db8139f..030dcd55963 100644 --- a/tardis/io/logger/tests/test_logging.py +++ b/tardis/io/logger/tests/test_logging.py @@ -3,7 +3,8 @@ from tardis.io.config_reader import Configuration from tardis.simulation import Simulation -from tardis import run_tardis, LOGGING_LEVELS +from tardis.io.logger.simulation_logger import LOGGING_LEVELS +from tardis import run_tardis def test_logging_simulation(atomic_data_fname, caplog): diff --git a/tardis/simulation/tests/test_simulation.py b/tardis/simulation/tests/test_simulation.py index f4c6ac9fdf8..88f3daff827 100644 --- a/tardis/simulation/tests/test_simulation.py +++ b/tardis/simulation/tests/test_simulation.py @@ -5,7 +5,7 @@ from tardis.io.config_reader import Configuration from tardis.simulation import Simulation -from tardis import run_tardis, LOGGING_LEVELS +from tardis import run_tardis import numpy as np import pandas as pd