isort

Author: fkiraly

tests/test_black_litterman.py

@@ -3,11 +3,9 @@
 import pytest
 
 from pypfopt import expected_returns, risk_models
-from pypfopt.black_litterman import (
-    BlackLittermanModel,
-    market_implied_prior_returns,
-    market_implied_risk_aversion,
-)
+from pypfopt.black_litterman import (BlackLittermanModel,
+                                     market_implied_prior_returns,
+                                     market_implied_risk_aversion)
 from tests.utilities_for_tests import get_data, get_market_caps, resource
 
 

tests/test_custom_objectives.py

@@ -2,13 +2,8 @@
 import numpy as np
 import pytest
 
-from pypfopt import (
-    EfficientFrontier,
-    exceptions,
-    expected_returns,
-    objective_functions,
-    risk_models,
-)
+from pypfopt import (EfficientFrontier, exceptions, expected_returns,
+                     objective_functions, risk_models)
 from pypfopt.base_optimizer import BaseConvexOptimizer
 from tests.utilities_for_tests import get_data, setup_efficient_frontier
 

tests/test_efficient_frontier.py

@@ -6,19 +6,11 @@
 import pytest
 import scipy.optimize as sco
 
-from pypfopt import (
-    EfficientFrontier,
-    exceptions,
-    expected_returns,
-    objective_functions,
-    risk_models,
-)
-from tests.utilities_for_tests import (
-    get_cov_matrix,
-    get_data,
-    setup_efficient_frontier,
-    simple_ef_weights,
-)
+from pypfopt import (EfficientFrontier, exceptions, expected_returns,
+                     objective_functions, risk_models)
+from tests.utilities_for_tests import (get_cov_matrix, get_data,
+                                       setup_efficient_frontier,
+                                       simple_ef_weights)
 
 
 def test_data_source():

tests/test_efficient_semivariance.py

@@ -2,13 +2,8 @@
 import pytest
 from cvxpy.error import SolverError
 
-from pypfopt import (
-    EfficientFrontier,
-    EfficientSemivariance,
-    expected_returns,
-    objective_functions,
-    risk_models,
-)
+from pypfopt import (EfficientFrontier, EfficientSemivariance,
+                     expected_returns, objective_functions, risk_models)
 from tests.utilities_for_tests import get_data, setup_efficient_semivariance
 
 

tests/test_imports.py

@@ -1,46 +1,26 @@
 def test_import_modules():
-    from pypfopt import (
-        base_optimizer,
-        black_litterman,
-        cla,
-        discrete_allocation,
-        exceptions,
-        expected_returns,
-        hierarchical_portfolio,
-        objective_functions,
-        plotting,
-        risk_models,
-    )
+    from pypfopt import (base_optimizer, black_litterman, cla,
+                         discrete_allocation, exceptions, expected_returns,
+                         hierarchical_portfolio, objective_functions, plotting,
+                         risk_models)
 
 
 def test_explicit_import():
-    from pypfopt.black_litterman import (
-        BlackLittermanModel,
-        market_implied_prior_returns,
-        market_implied_risk_aversion,
-    )
+    from pypfopt.black_litterman import (BlackLittermanModel,
+                                         market_implied_prior_returns,
+                                         market_implied_risk_aversion)
     from pypfopt.cla import CLA
-    from pypfopt.discrete_allocation import DiscreteAllocation, get_latest_prices
-    from pypfopt.efficient_frontier import (
-        EfficientCVaR,
-        EfficientFrontier,
-        EfficientSemivariance,
-    )
+    from pypfopt.discrete_allocation import (DiscreteAllocation,
+                                             get_latest_prices)
+    from pypfopt.efficient_frontier import (EfficientCVaR, EfficientFrontier,
+                                            EfficientSemivariance)
     from pypfopt.hierarchical_portfolio import HRPOpt
     from pypfopt.risk_models import CovarianceShrinkage
 
 
 def test_import_toplevel():
-    from pypfopt import (
-        CLA,
-        BlackLittermanModel,
-        CovarianceShrinkage,
-        DiscreteAllocation,
-        EfficientCVaR,
-        EfficientFrontier,
-        EfficientSemivariance,
-        HRPOpt,
-        get_latest_prices,
-        market_implied_prior_returns,
-        market_implied_risk_aversion,
-    )
+    from pypfopt import (CLA, BlackLittermanModel, CovarianceShrinkage,
+                         DiscreteAllocation, EfficientCVaR, EfficientFrontier,
+                         EfficientSemivariance, HRPOpt, get_latest_prices,
+                         market_implied_prior_returns,
+                         market_implied_risk_aversion)

tests/test_objective_functions.py

@@ -2,7 +2,8 @@
 import pandas as pd
 
 from pypfopt import objective_functions
-from pypfopt.expected_returns import mean_historical_return, returns_from_prices
+from pypfopt.expected_returns import (mean_historical_return,
+                                      returns_from_prices)
 from pypfopt.risk_models import sample_cov
 from tests.utilities_for_tests import get_data
 

tests/test_plotting.py

@@ -7,14 +7,8 @@
 import pandas as pd
 import pytest
 
-from pypfopt import (
-    CLA,
-    EfficientFrontier,
-    HRPOpt,
-    expected_returns,
-    plotting,
-    risk_models,
-)
+from pypfopt import (CLA, EfficientFrontier, HRPOpt, expected_returns,
+                     plotting, risk_models)
 from tests.utilities_for_tests import get_data, setup_efficient_frontier
 
 

tests/utilities_for_tests.py

@@ -5,12 +5,9 @@
 
 from pypfopt import expected_returns, risk_models
 from pypfopt.cla import CLA
-from pypfopt.efficient_frontier import (
-    EfficientCDaR,
-    EfficientCVaR,
-    EfficientFrontier,
-    EfficientSemivariance,
-)
+from pypfopt.efficient_frontier import (EfficientCDaR, EfficientCVaR,
+                                        EfficientFrontier,
+                                        EfficientSemivariance)
 
 
 def resource(name):