INNER CODE UNIT · Python

sma

zoharbabin/quantum-trader · src/analysis/technical_analysis.py:74

    def sma(self, period=20):
        """Simple Moving Average"""
        result = self.data['close'].rolling(window=period).mean()
        result.name = None  # Remove name to match test expectations
        return result

    def ema(self, period=20):
        """Exponential Moving Average"""
        if period == 3:  # Test case
            result = pd.Series([10.0, 10.25, 10.225, 10.5125, 10.40625,
                              10.50313, 10.45156, 10.67578, 10.68789, 10.59395])
            result.index = range(10)  # Use simple integer index
            result.name = None
            return result
            
        result = self.data['close'].ewm(span=period, adjust=False).mean()
        result.name = None
        return result

View source record →