File trajpy.py has 414 lines of code (exceeds 250 allowed). Consider refactoring.

Open

import numpy as np
from scipy.stats import linregress
import trajpy.auxiliar_functions as aux
from typing import Union, List, Dict, Tuple
import warnings

Function compute_selected has a Cognitive Complexity of 26 (exceeds 5 allowed). Consider refactoring.

Open

    def compute_selected(self) -> None:
        '''
            compute the selected features
        '''
        

• Read up
  Read up

Function __init__ has 28 lines of code (exceeds 25 allowed). Consider refactoring.

Open

    def __init__(self, trajectory: np.ndarray=np.zeros((1, 2)), box_length: int = None, **params) -> None:
        """
        Initialization function that can be left blank for using staticmethods.
        It can be initialized with an array with shape (N, dim)
        where dim is the number of spatial dimensions plus the time component.

Function normal_diffusion has a Cognitive Complexity of 9 (exceeds 5 allowed). Consider refactoring.

Open

def normal_diffusion(n_steps: int, n_samples: int, dx: float, y0: float, D: float, dt: float) -> Tuple[np.ndarray, np.ndarray]:
    """
    Generates an ensemble of normal diffusion trajectories.

    :param n_steps: total steps

• Read up
  Read up

Function confined_diffusion has a Cognitive Complexity of 6 (exceeds 5 allowed). Consider refactoring.

Open

def confined_diffusion(radius: float, n_steps: int, n_samples: int, dx: float, y0: float, D: float, dt: float) -> Tuple[np.ndarray, np.ndarray]:
    """
    Generates trajectories under confinement.

    :param radius: confinement radius

• Read up
  Read up