From d7ca9acbb668d4c73f07eddf0278c08bbdae7be7 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Mon, 6 Jul 2020 11:47:05 +0200 Subject: Move ParamFit, PTAModel, AnalyticModel to model.py module --- test/test_parameters.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test/test_parameters.py') diff --git a/test/test_parameters.py b/test/test_parameters.py index 22efccf..e36b1a1 100755 --- a/test/test_parameters.py +++ b/test/test_parameters.py @@ -1,9 +1,9 @@ #!/usr/bin/env python3 -from dfatool import dfatool as dt from dfatool import parameters from dfatool.utils import by_name_to_by_param from dfatool.functions import analytic +from dfatool.model import ParallelParamFit import unittest import numpy as np @@ -52,7 +52,7 @@ class TestModels(unittest.TestCase): # Fit individual functions for each parameter (only "p_linear" in this case) - paramfit = dt.ParallelParamFit(by_param) + paramfit = ParallelParamFit(by_param) paramfit.enqueue("TX", "power", 1, "p_linear") paramfit.fit() @@ -133,7 +133,7 @@ class TestModels(unittest.TestCase): self.assertEqual(stats.depends_on_param("someKey", "ll", "log_inv"), True) self.assertEqual(stats.depends_on_param("someKey", "ll", "square_none"), False) - paramfit = dt.ParallelParamFit(by_param) + paramfit = ParallelParamFit(by_param) paramfit.enqueue("someKey", "lls", 0, "lin_lin") paramfit.enqueue("someKey", "lls", 1, "log_inv") paramfit.enqueue("someKey", "lls", 2, "square_none") -- cgit v1.2.3