diff --git a/examples/cluster_number_counts/compare_integration_methods.py b/examples/cluster_number_counts/compare_integration_methods.py index b8d216bce..7b3798e6b 100644 --- a/examples/cluster_number_counts/compare_integration_methods.py +++ b/examples/cluster_number_counts/compare_integration_methods.py @@ -1,4 +1,5 @@ """Test integral methods for cluster abundance.""" + import time import itertools diff --git a/examples/cosmicshear/cosmicshear.py b/examples/cosmicshear/cosmicshear.py index 4f243d033..ad73809bd 100644 --- a/examples/cosmicshear/cosmicshear.py +++ b/examples/cosmicshear/cosmicshear.py @@ -1,5 +1,6 @@ """Define the likelihood factory function for the cosmic shear example. """ + import os import sacc diff --git a/examples/cosmicshear/generate_cosmicshear_data.py b/examples/cosmicshear/generate_cosmicshear_data.py index aafdd2ca5..876b7efb6 100644 --- a/examples/cosmicshear/generate_cosmicshear_data.py +++ b/examples/cosmicshear/generate_cosmicshear_data.py @@ -1,5 +1,6 @@ """ Generate the cosmicshear.fits file. """ + import numpy as np import sacc import pyccl as ccl diff --git a/examples/des_y1_3x2pt/generate_des_data.py b/examples/des_y1_3x2pt/generate_des_data.py index 0aa7d9b3e..0f21d6cd9 100644 --- a/examples/des_y1_3x2pt/generate_des_data.py +++ b/examples/des_y1_3x2pt/generate_des_data.py @@ -5,6 +5,7 @@ http://desdr-server.ncsa.illinois.edu/despublic/y1a1_files/chains/2pt_NG_mcal_1110.fits """ + from typing import Dict, Tuple import fitsio import numpy as np diff --git a/examples/srd_sn/generate_sn_data.py b/examples/srd_sn/generate_sn_data.py index 3bfde3b08..c47dbf74c 100644 --- a/examples/srd_sn/generate_sn_data.py +++ b/examples/srd_sn/generate_sn_data.py @@ -1,5 +1,6 @@ """Generate SACC data into file srd-y1-converted.sacc. """ + import os import tarfile import urllib.request diff --git a/examples/srd_sn/sn_srd.py b/examples/srd_sn/sn_srd.py index 9dbb0dea2..331fc16c4 100644 --- a/examples/srd_sn/sn_srd.py +++ b/examples/srd_sn/sn_srd.py @@ -1,5 +1,6 @@ """Demonstration of the use of the :class:`Supernova` statistics object. """ + import sacc import firecrown.likelihood.gauss_family.statistic.supernova as sn from firecrown.likelihood.gauss_family.gaussian import ConstGaussian diff --git a/firecrown/connector/cobaya/__init__.py b/firecrown/connector/cobaya/__init__.py index 8dabb33d3..7f233234c 100644 --- a/firecrown/connector/cobaya/__init__.py +++ b/firecrown/connector/cobaya/__init__.py @@ -5,5 +5,4 @@ """ - # flake8: noqa diff --git a/firecrown/connector/cobaya/ccl.py b/firecrown/connector/cobaya/ccl.py index 31f86a14a..e2e41d0cb 100644 --- a/firecrown/connector/cobaya/ccl.py +++ b/firecrown/connector/cobaya/ccl.py @@ -4,6 +4,7 @@ Provide the class CCLConnector, which is an implementation of a Cobaya Theory. """ + from __future__ import annotations from typing import Optional, Dict, List, Union diff --git a/firecrown/connector/cobaya/likelihood.py b/firecrown/connector/cobaya/likelihood.py index 0ea2bef16..65d22ad85 100644 --- a/firecrown/connector/cobaya/likelihood.py +++ b/firecrown/connector/cobaya/likelihood.py @@ -5,6 +5,7 @@ This module provides the class :class:`LikelihoodConnector`, which is an implementation of a Cobaya likelihood. """ + from typing import List, Dict, Union import numpy as np diff --git a/firecrown/connector/cosmosis/__init__.py b/firecrown/connector/cosmosis/__init__.py index e0e5aa406..62270211e 100644 --- a/firecrown/connector/cosmosis/__init__.py +++ b/firecrown/connector/cosmosis/__init__.py @@ -5,5 +5,4 @@ """ - # flake8: noqa diff --git a/firecrown/connector/cosmosis/likelihood.py b/firecrown/connector/cosmosis/likelihood.py index c77a6cbee..3a027c5e2 100644 --- a/firecrown/connector/cosmosis/likelihood.py +++ b/firecrown/connector/cosmosis/likelihood.py @@ -7,6 +7,7 @@ likelihood abstract base class; it the implementation of a CosmoSIS module, not a specific likelihood. """ + import cosmosis.datablock from cosmosis.datablock import option_section from cosmosis.datablock import names as section_names diff --git a/firecrown/likelihood/__init__.py b/firecrown/likelihood/__init__.py index 6a142daba..636d36e76 100644 --- a/firecrown/likelihood/__init__.py +++ b/firecrown/likelihood/__init__.py @@ -6,5 +6,4 @@ """ - # flake8: noqa diff --git a/firecrown/likelihood/gauss_family/statistic/__init__.py b/firecrown/likelihood/gauss_family/statistic/__init__.py index 874b8f627..a9fb3bdaa 100644 --- a/firecrown/likelihood/gauss_family/statistic/__init__.py +++ b/firecrown/likelihood/gauss_family/statistic/__init__.py @@ -1,4 +1,5 @@ """This package provides statistics objects to be used by Gaussian Family likelihoods. """ + # flake8: noqa diff --git a/firecrown/likelihood/gauss_family/statistic/binned_cluster_number_counts.py b/firecrown/likelihood/gauss_family/statistic/binned_cluster_number_counts.py index 305f4b5d6..234657d28 100644 --- a/firecrown/likelihood/gauss_family/statistic/binned_cluster_number_counts.py +++ b/firecrown/likelihood/gauss_family/statistic/binned_cluster_number_counts.py @@ -3,6 +3,7 @@ The binned cluster number counts statistic predicts the number of galaxy clusters within a single redshift and mass bin. """ + from __future__ import annotations from typing import List, Optional diff --git a/firecrown/likelihood/gauss_family/statistic/source/__init__.py b/firecrown/likelihood/gauss_family/statistic/source/__init__.py index a2eeac361..b57beda53 100644 --- a/firecrown/likelihood/gauss_family/statistic/source/__init__.py +++ b/firecrown/likelihood/gauss_family/statistic/source/__init__.py @@ -1,4 +1,5 @@ """This package contains the base class :class:`Source` for :class:`TwoPoint` sources and implementations. """ + # flake8: noqa diff --git a/firecrown/modeling_tools.py b/firecrown/modeling_tools.py index 62521e675..bee5af689 100644 --- a/firecrown/modeling_tools.py +++ b/firecrown/modeling_tools.py @@ -4,6 +4,7 @@ :class:`pyccl.Cosmology` class. This is used by likelihoods that need to access reusable objects, such as perturbation theory or halo model calculators. """ + from __future__ import annotations from typing import Dict, Optional, Collection from abc import ABC, abstractmethod diff --git a/firecrown/models/__init__.py b/firecrown/models/__init__.py index 947d0d7b0..ab71fab69 100644 --- a/firecrown/models/__init__.py +++ b/firecrown/models/__init__.py @@ -2,5 +2,4 @@ """ - # flake8: noqa diff --git a/firecrown/models/cluster/abundance.py b/firecrown/models/cluster/abundance.py index 8c9b9a511..8d282cc34 100644 --- a/firecrown/models/cluster/abundance.py +++ b/firecrown/models/cluster/abundance.py @@ -4,6 +4,7 @@ and phenomenological predictions. This module contains the classes and functions that produce those predictions. """ + from typing import Dict, Tuple import numpy as np import numpy.typing as npt diff --git a/firecrown/models/cluster/abundance_data.py b/firecrown/models/cluster/abundance_data.py index a37975952..cd63fca4d 100644 --- a/firecrown/models/cluster/abundance_data.py +++ b/firecrown/models/cluster/abundance_data.py @@ -1,5 +1,6 @@ """The module responsible for extracting cluster data from a sacc file. """ + from typing import Tuple, List import numpy as np import numpy.typing as npt diff --git a/firecrown/models/cluster/integrator/integrator.py b/firecrown/models/cluster/integrator/integrator.py index 7620623fc..72aaf0bf1 100644 --- a/firecrown/models/cluster/integrator/integrator.py +++ b/firecrown/models/cluster/integrator/integrator.py @@ -3,6 +3,7 @@ This module holds the classes that define the interface required to integrate a function. """ + from abc import ABC, abstractmethod from typing import Callable, List, Tuple diff --git a/firecrown/models/cluster/integrator/numcosmo_integrator.py b/firecrown/models/cluster/integrator/numcosmo_integrator.py index e457620b7..f97cdf31d 100644 --- a/firecrown/models/cluster/integrator/numcosmo_integrator.py +++ b/firecrown/models/cluster/integrator/numcosmo_integrator.py @@ -2,6 +2,7 @@ This module holds the NumCosmo implementation of the integrator classes """ + from enum import Enum from typing import Callable, Optional, Tuple diff --git a/firecrown/models/cluster/integrator/scipy_integrator.py b/firecrown/models/cluster/integrator/scipy_integrator.py index 4869f60c9..bc609dd34 100644 --- a/firecrown/models/cluster/integrator/scipy_integrator.py +++ b/firecrown/models/cluster/integrator/scipy_integrator.py @@ -2,6 +2,7 @@ This module holds the scipy implementation of the integrator classes """ + from typing import Callable import numpy as np diff --git a/firecrown/models/cluster/kernel.py b/firecrown/models/cluster/kernel.py index df3fcd08f..875389a83 100644 --- a/firecrown/models/cluster/kernel.py +++ b/firecrown/models/cluster/kernel.py @@ -2,6 +2,7 @@ This module holds the classes that define the kernels that can be included in the cluster abundance integrand.""" + from enum import Enum from typing import Tuple import numpy.typing as npt diff --git a/firecrown/models/cluster/mass_proxy.py b/firecrown/models/cluster/mass_proxy.py index 54170f7d6..2c11408d2 100644 --- a/firecrown/models/cluster/mass_proxy.py +++ b/firecrown/models/cluster/mass_proxy.py @@ -3,6 +3,7 @@ This module holds the classes that define the mass richness relations that can be included in the cluster abundance integrand. These are implementations of Kernels.""" + from abc import abstractmethod from typing import Tuple diff --git a/firecrown/models/cluster/properties.py b/firecrown/models/cluster/properties.py index e141b13b9..3e6fa8bf2 100644 --- a/firecrown/models/cluster/properties.py +++ b/firecrown/models/cluster/properties.py @@ -1,4 +1,5 @@ """Module containing classes relevant to defining cluster properties.""" + from enum import Flag, auto diff --git a/firecrown/models/cluster/recipes/cluster_recipe.py b/firecrown/models/cluster/recipes/cluster_recipe.py index 4de2bf496..29f09b1eb 100644 --- a/firecrown/models/cluster/recipes/cluster_recipe.py +++ b/firecrown/models/cluster/recipes/cluster_recipe.py @@ -1,4 +1,5 @@ """Module for defining the ClusterRecipe class""" + from abc import ABC, abstractmethod from typing import Optional diff --git a/firecrown/models/cluster/recipes/murata_binned_spec_z.py b/firecrown/models/cluster/recipes/murata_binned_spec_z.py index bdfbf9ff0..12d22e0d9 100644 --- a/firecrown/models/cluster/recipes/murata_binned_spec_z.py +++ b/firecrown/models/cluster/recipes/murata_binned_spec_z.py @@ -1,4 +1,5 @@ """Module for defining the classes used in the MurataBinnedSpecZ cluster recipe.""" + from typing import Callable, Optional, Tuple import numpy as np diff --git a/firecrown/models/cluster/recipes/murata_unbinned_spec_z.py b/firecrown/models/cluster/recipes/murata_unbinned_spec_z.py index 472a9b346..f1a82d04a 100644 --- a/firecrown/models/cluster/recipes/murata_unbinned_spec_z.py +++ b/firecrown/models/cluster/recipes/murata_unbinned_spec_z.py @@ -1,4 +1,5 @@ """Module for defining the classes used in the MurataUnbinnedSpecZ cluster recipe.""" + from typing import Callable, Optional import numpy as np diff --git a/firecrown/updatable.py b/firecrown/updatable.py index ecb68c28c..33004f675 100644 --- a/firecrown/updatable.py +++ b/firecrown/updatable.py @@ -300,7 +300,6 @@ def _get_derived_parameters(self) -> DerivedParameterCollection: class UpdatableCollection(UserList[Any]): - """UpdatableCollection is a list of Updatable objects and is itself supports :meth:`update` and :meth:`reset` (although it does not inherit from :class:`Updatable`). diff --git a/tests/cluster_recipes/test_murata_binned_spec_z.py b/tests/cluster_recipes/test_murata_binned_spec_z.py index 22ebb6d5f..ddb84202b 100644 --- a/tests/cluster_recipes/test_murata_binned_spec_z.py +++ b/tests/cluster_recipes/test_murata_binned_spec_z.py @@ -1,4 +1,5 @@ """Tests for the cluster abundance module.""" + from unittest.mock import Mock import numpy as np diff --git a/tests/cluster_recipes/test_murata_unbinned_spec_z.py b/tests/cluster_recipes/test_murata_unbinned_spec_z.py index 76207c140..938001fa4 100644 --- a/tests/cluster_recipes/test_murata_unbinned_spec_z.py +++ b/tests/cluster_recipes/test_murata_unbinned_spec_z.py @@ -1,4 +1,5 @@ """Tests for the cluster abundance module.""" + from unittest.mock import Mock import numpy as np diff --git a/tests/connector/cosmosis/test_cosmosis_module.py b/tests/connector/cosmosis/test_cosmosis_module.py index c3fbf655c..31460d862 100644 --- a/tests/connector/cosmosis/test_cosmosis_module.py +++ b/tests/connector/cosmosis/test_cosmosis_module.py @@ -3,6 +3,7 @@ As a unit test, what this can test is very limited. This test do not invoke the `cosmosis` executable. """ + from os.path import expandvars import yaml import pytest diff --git a/tests/connector/test_mapping.py b/tests/connector/test_mapping.py index f14fd53cc..fa6f5245a 100644 --- a/tests/connector/test_mapping.py +++ b/tests/connector/test_mapping.py @@ -1,6 +1,7 @@ """ Unit testing for the mapping module. """ + from typing import Any, Dict import pytest import numpy as np diff --git a/tests/likelihood/gauss_family/statistic/source/test_source.py b/tests/likelihood/gauss_family/statistic/source/test_source.py index f041fd03f..e04ee935b 100644 --- a/tests/likelihood/gauss_family/statistic/source/test_source.py +++ b/tests/likelihood/gauss_family/statistic/source/test_source.py @@ -1,6 +1,7 @@ """ Tests for the module firecrown.likelihood.gauss_family.statistic.source.source. """ + import pytest import numpy as np diff --git a/tests/likelihood/gauss_family/statistic/test_cluster_number_counts.py b/tests/likelihood/gauss_family/statistic/test_cluster_number_counts.py index cf542d142..aaf3f3c9c 100644 --- a/tests/likelihood/gauss_family/statistic/test_cluster_number_counts.py +++ b/tests/likelihood/gauss_family/statistic/test_cluster_number_counts.py @@ -1,4 +1,5 @@ """Tests for the binned cluster number counts module.""" + from unittest.mock import Mock import sacc import pytest diff --git a/tests/likelihood/gauss_family/statistic/test_statistic.py b/tests/likelihood/gauss_family/statistic/test_statistic.py index e4153c0a7..904954e0d 100644 --- a/tests/likelihood/gauss_family/statistic/test_statistic.py +++ b/tests/likelihood/gauss_family/statistic/test_statistic.py @@ -1,6 +1,7 @@ """ Tests for the module firecrown.likelihood.gauss_family.statistic.statistic. """ + from typing import List import numpy as np import pytest diff --git a/tests/likelihood/gauss_family/statistic/test_supernova.py b/tests/likelihood/gauss_family/statistic/test_supernova.py index 38577c6e5..fe6cf7e19 100644 --- a/tests/likelihood/gauss_family/statistic/test_supernova.py +++ b/tests/likelihood/gauss_family/statistic/test_supernova.py @@ -1,5 +1,6 @@ """Tests for the Supernova statistic. """ + import pytest import sacc diff --git a/tests/likelihood/gauss_family/statistic/test_two_point.py b/tests/likelihood/gauss_family/statistic/test_two_point.py index 9c7e381a3..5e40fb257 100644 --- a/tests/likelihood/gauss_family/statistic/test_two_point.py +++ b/tests/likelihood/gauss_family/statistic/test_two_point.py @@ -1,6 +1,7 @@ """ Tests for the TwoPoint module. """ + import numpy as np import pytest diff --git a/tests/likelihood/lkdir/lk_derived_parameter.py b/tests/likelihood/lkdir/lk_derived_parameter.py index baff91fa7..9b490f89e 100644 --- a/tests/likelihood/lkdir/lk_derived_parameter.py +++ b/tests/likelihood/lkdir/lk_derived_parameter.py @@ -2,6 +2,7 @@ Provides a trivial likelihood factory function for testing purposes. The likelihood created provides one derived parameter named "derived_param0". """ + from firecrown.likelihood.likelihood import NamedParameters from . import lkmodule diff --git a/tests/likelihood/lkdir/lk_needing_param.py b/tests/likelihood/lkdir/lk_needing_param.py index f1ce67679..e228cfdd4 100644 --- a/tests/likelihood/lkdir/lk_needing_param.py +++ b/tests/likelihood/lkdir/lk_needing_param.py @@ -2,6 +2,7 @@ Provides a trivial likelihood factory function for testing purposes. The likelihood created requires a string parameter named "sacc_file". """ + from firecrown.likelihood.likelihood import NamedParameters from . import lkmodule diff --git a/tests/likelihood/lkdir/lk_sampler_parameter.py b/tests/likelihood/lkdir/lk_sampler_parameter.py index 32a3772b9..7cf9a2d37 100644 --- a/tests/likelihood/lkdir/lk_sampler_parameter.py +++ b/tests/likelihood/lkdir/lk_sampler_parameter.py @@ -3,6 +3,7 @@ The likelihood created requires a string parameter named "parameter_prefix" and has a sampler parameter named "sampler_param0". """ + from firecrown.likelihood.likelihood import NamedParameters from . import lkmodule diff --git a/tests/likelihood/lkdir/lkmodule.py b/tests/likelihood/lkdir/lkmodule.py index c83925395..4c804f721 100644 --- a/tests/likelihood/lkdir/lkmodule.py +++ b/tests/likelihood/lkdir/lkmodule.py @@ -1,6 +1,7 @@ """ Provides a trivial likelihood class and factory function for testing purposes. """ + import sacc from firecrown.parameters import DerivedParameterCollection, DerivedParameterScalar from firecrown.likelihood.likelihood import Likelihood, NamedParameters diff --git a/tests/likelihood/lkdir/lkscript.py b/tests/likelihood/lkdir/lkscript.py index c9e65918f..011abc77d 100644 --- a/tests/likelihood/lkdir/lkscript.py +++ b/tests/likelihood/lkdir/lkscript.py @@ -1,6 +1,7 @@ """ Provides a trivial likelihood factory function for testing purposes. """ + from firecrown.modeling_tools import ModelingTools from . import lkmodule diff --git a/tests/likelihood/lkdir/lkscript_invalid.py b/tests/likelihood/lkdir/lkscript_invalid.py index 02aece5b3..3e1197f7c 100644 --- a/tests/likelihood/lkdir/lkscript_invalid.py +++ b/tests/likelihood/lkdir/lkscript_invalid.py @@ -4,6 +4,7 @@ It should raise an exception because the factory function does not define a build_likelihood Callable. """ + from . import lkmodule diff --git a/tests/likelihood/lkdir/lkscript_not_a_function.py b/tests/likelihood/lkdir/lkscript_not_a_function.py index 95bd5c30f..db963591c 100644 --- a/tests/likelihood/lkdir/lkscript_not_a_function.py +++ b/tests/likelihood/lkdir/lkscript_not_a_function.py @@ -5,5 +5,4 @@ a build_likelihood as a Callable. """ - build_likelihood = "I am not a function" diff --git a/tests/likelihood/lkdir/lkscript_old.py b/tests/likelihood/lkdir/lkscript_old.py index bf2983604..ed5030749 100644 --- a/tests/likelihood/lkdir/lkscript_old.py +++ b/tests/likelihood/lkdir/lkscript_old.py @@ -1,6 +1,7 @@ """ Provides a trivial likelihood factory function for testing purposes. """ + from . import lkmodule diff --git a/tests/likelihood/lkdir/lkscript_returns_wrong_type_tools.py b/tests/likelihood/lkdir/lkscript_returns_wrong_type_tools.py index 1ce1abe44..81b84b4e2 100644 --- a/tests/likelihood/lkdir/lkscript_returns_wrong_type_tools.py +++ b/tests/likelihood/lkdir/lkscript_returns_wrong_type_tools.py @@ -4,6 +4,7 @@ It should raise an exception because the factory function does not return a Likelihood object. """ + from . import lkmodule diff --git a/tests/likelihood/test_cosmosis_likelihood.py b/tests/likelihood/test_cosmosis_likelihood.py index 2f4d4cfd2..1923b1a27 100644 --- a/tests/likelihood/test_cosmosis_likelihood.py +++ b/tests/likelihood/test_cosmosis_likelihood.py @@ -1,4 +1,5 @@ """Tests for the CosmoSIS likelihood module.""" + from pytest import fixture from cosmosis.datablock import DataBlock import firecrown.connector.cosmosis.likelihood as like diff --git a/tests/likelihood/test_likelihood.py b/tests/likelihood/test_likelihood.py index 015f813c5..f811d6064 100644 --- a/tests/likelihood/test_likelihood.py +++ b/tests/likelihood/test_likelihood.py @@ -1,6 +1,7 @@ """ Tests for the module firecrown.likelihood.likelihood. """ + import os import sys import pytest diff --git a/tests/likelihood/test_named_parameters.py b/tests/likelihood/test_named_parameters.py index bdd3f31c4..0b5ee1a68 100644 --- a/tests/likelihood/test_named_parameters.py +++ b/tests/likelihood/test_named_parameters.py @@ -1,6 +1,5 @@ """Tests for the class firecrown.likelihood.NamedParameters.""" - import pytest import numpy as np diff --git a/tests/test_cluster_abundance.py b/tests/test_cluster_abundance.py index 7dd171782..2f5e3fb26 100644 --- a/tests/test_cluster_abundance.py +++ b/tests/test_cluster_abundance.py @@ -1,4 +1,5 @@ """Tests for the cluster abundance module.""" + import numpy as np import pyccl import pytest diff --git a/tests/test_cluster_binning.py b/tests/test_cluster_binning.py index ad6656039..5991fb3ba 100644 --- a/tests/test_cluster_binning.py +++ b/tests/test_cluster_binning.py @@ -1,4 +1,5 @@ """Tests for the cluster binning module""" + from unittest.mock import Mock import pytest diff --git a/tests/test_cluster_data.py b/tests/test_cluster_data.py index 4c6084902..536848cf6 100644 --- a/tests/test_cluster_data.py +++ b/tests/test_cluster_data.py @@ -1,4 +1,5 @@ """Tests for the cluster abundance data module.""" + import pytest import sacc from firecrown.models.cluster.abundance_data import AbundanceData diff --git a/tests/test_cluster_kernels.py b/tests/test_cluster_kernels.py index 749426b4e..24a3e3e96 100644 --- a/tests/test_cluster_kernels.py +++ b/tests/test_cluster_kernels.py @@ -1,4 +1,5 @@ """Tests for the cluster kernel module.""" + import numpy as np import pytest diff --git a/tests/test_cluster_mass_richness.py b/tests/test_cluster_mass_richness.py index 840c99ca0..eaf37ad13 100644 --- a/tests/test_cluster_mass_richness.py +++ b/tests/test_cluster_mass_richness.py @@ -1,4 +1,5 @@ """Tests for the cluster mass richness module.""" + import pytest import numpy as np from scipy.integrate import quad diff --git a/tests/test_descriptors.py b/tests/test_descriptors.py index edbb27dba..d69ab369a 100644 --- a/tests/test_descriptors.py +++ b/tests/test_descriptors.py @@ -1,6 +1,7 @@ """ Tests for the module firecrown.descriptors. """ + import math import pytest from firecrown.descriptors import TypeFloat, TypeString diff --git a/tests/test_modeling_tools.py b/tests/test_modeling_tools.py index 081669135..036585f89 100644 --- a/tests/test_modeling_tools.py +++ b/tests/test_modeling_tools.py @@ -1,6 +1,7 @@ """ Tests for the module firecrown.modeling_tools """ + import pytest import pyccl from firecrown.modeling_tools import ModelingTools, PowerspectrumModifier diff --git a/tests/test_parameters.py b/tests/test_parameters.py index c75e08d94..1525a3adc 100644 --- a/tests/test_parameters.py +++ b/tests/test_parameters.py @@ -1,6 +1,7 @@ """ Tests for the module firecrown.parameters. """ + import pytest import numpy as np from firecrown.parameters import RequiredParameters, parameter_get_full_name, ParamsMap diff --git a/tests/test_pt_systematics.py b/tests/test_pt_systematics.py index 701add600..22ae01cb2 100644 --- a/tests/test_pt_systematics.py +++ b/tests/test_pt_systematics.py @@ -2,6 +2,7 @@ Tests for the perturbation theory systematics for both Weak Lensing and Number Counts. """ + import os import pytest diff --git a/tests/test_pyccl.py b/tests/test_pyccl.py index 0657bee78..703083882 100644 --- a/tests/test_pyccl.py +++ b/tests/test_pyccl.py @@ -2,6 +2,7 @@ Test a few of the features of pyccl upon which we rely, but which might change in future release of pyccl. """ + import pytest import pyccl diff --git a/tests/test_updatable.py b/tests/test_updatable.py index 1e985b05a..a28da1ea3 100644 --- a/tests/test_updatable.py +++ b/tests/test_updatable.py @@ -1,6 +1,7 @@ """ Tests for the Updatable class. """ + from itertools import permutations import pytest import numpy as np diff --git a/tests/test_utils.py b/tests/test_utils.py index 33a039edf..5086a80d1 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -1,6 +1,7 @@ """ Tests for the firecrown.utils modle. """ + from firecrown.utils import upper_triangle_indices diff --git a/tests/test_version.py b/tests/test_version.py index 1a62191ef..c374b58b9 100644 --- a/tests/test_version.py +++ b/tests/test_version.py @@ -3,6 +3,7 @@ Note that this test intentionally has to be adjusted for each new release of firecrown. """ + import firecrown