inewave 1.6.0__py3-none-any.whl → 1.7.1__py3-none-any.whl

tests/newave/test_pmo.py

@@ -8,6 +8,7 @@ from inewave.newave.modelos.pmo import (
     BlocoCustoOperacaoPMO,
     BlocoCustoOperacaoTotalPMO,
     BlocoProdutibilidadesConfiguracaoPMO,
+    BlocoEnergiaArmazenadaMaximaPMO,
     BlocoEnergiaArmazenadaInicialPMO,
     BlocoVolumeArmazenadoInicialPMO,
     BlocoPenalidadeViolacaoOutrosUsosPMO,
@@ -17,6 +18,8 @@ from inewave.newave.modelos.pmo import (
     BlocoPenalidadeViolacaoEvaporacaoPMO,
     BlocoPenalidadeViolacaoTurbinamentoMaximoPMO,
     BlocoPenalidadeViolacaoTurbinamentoMinimoPMO,
+    BlocoGeracaoMinimaUsinasTermicasPMO,
+    BlocoGeracaoMaximaUsinasTermicasPMO,
 )
 
 from inewave.newave import Pmo
@@ -37,6 +40,7 @@ from tests.mocks.arquivos.pmo import MockBlocoMARSPMOFinal
 from tests.mocks.arquivos.pmo import MockBlocoRiscoDeficitENSPMO
 from tests.mocks.arquivos.pmo import MockBlocoCustoOperacaoPMO
 from tests.mocks.arquivos.pmo import MockBlocoCustoOperacaoTotalPMO
+from tests.mocks.arquivos.pmo import MockBlocoEnergiaArmazenadaMaximaPMO
 from tests.mocks.arquivos.pmo import MockBlocoEnergiaArmazenadaInicialPMO
 from tests.mocks.arquivos.pmo import MockBlocoVolumeArmazenadoInicialPMO
 from tests.mocks.arquivos.pmo import MockPenalidadeViolacaoOutrosUsosPMO
@@ -50,6 +54,10 @@ from tests.mocks.arquivos.pmo import (
 from tests.mocks.arquivos.pmo import (
     MockPenalidadeViolacaoTurbinamentoMaximoPMO,
 )
+from tests.mocks.arquivos.pmo import (
+    MockBlocoGeracaoMinimaUsinasTermicasPMO,
+    MockBlocoGeracaoMaximaUsinasTermicasPMO,
+)
 from tests.mocks.arquivos.pmo import MockPMO
 
 ARQ_TESTE = "./tests/mocks/arquivos/__init__.py"
@@ -85,6 +93,23 @@ def test_eafpast_cfuga_medio():
     assert b.data.iloc[-1, -1] == 409.13
 
 
+def test_energia_armazenada_maxima():
+    m: MagicMock = mock_open(
+        read_data="".join(MockBlocoEnergiaArmazenadaMaximaPMO)
+    )
+    b = BlocoEnergiaArmazenadaMaximaPMO()
+    with patch("builtins.open", m):
+        with open("", "") as fp:
+            b.read(fp)
+
+    assert b.data.shape[0] == 720
+    assert b.data.shape[1] == 3
+    assert b.data.iloc[0, 0] == "SUDESTE"
+    assert b.data.iloc[0, 1] == 1
+    assert b.data.iloc[0, 2] == 50969.1
+    assert b.data.iloc[-1, -1] == 786.0
+
+
 def test_energia_armazenada_inicial():
     m: MagicMock = mock_open(
         read_data="".join(MockBlocoEnergiaArmazenadaInicialPMO)
@@ -319,6 +344,34 @@ def test_leitura_penalidades_violacao_turbinamento_maximo():
     assert b.data.iloc[-1, -1] == 842.25
 
 
+def test_leitura_geracao_termica_minima_usina():
+    m: MagicMock = mock_open(
+        read_data="".join(MockBlocoGeracaoMinimaUsinasTermicasPMO)
+    )
+    b = BlocoGeracaoMinimaUsinasTermicasPMO()
+    with patch("builtins.open", m):
+        with open("", "") as fp:
+            b.read(fp)
+
+    assert b.data.shape == (420, 4)
+    assert b.data.iloc[-1, -2] == 0.00
+    assert b.data.iloc[-1, -1] == datetime(2027, 12, 1)
+
+
+def test_leitura_geracao_termica_maxima_usina():
+    m: MagicMock = mock_open(
+        read_data="".join(MockBlocoGeracaoMaximaUsinasTermicasPMO)
+    )
+    b = BlocoGeracaoMaximaUsinasTermicasPMO()
+    with patch("builtins.open", m):
+        with open("", "") as fp:
+            b.read(fp)
+
+    assert b.data.shape == (60, 4)
+    assert b.data.iloc[-1, -2] == 558.02
+    assert b.data.iloc[-1, -1] == datetime(2027, 12, 1)
+
+
 def test_atributos_encontrados_pmo():
     m: MagicMock = mock_open(read_data="".join(MockPMO))
     with patch("builtins.open", m):
@@ -338,6 +391,7 @@ def test_atributos_encontrados_pmo():
         assert pmo.custo_operacao_total is not None
         assert pmo.desvio_custo_operacao_total is not None
         assert pmo.produtibilidades_equivalentes is not None
+        assert pmo.energia_armazenada_maxima is not None
         assert pmo.energia_armazenada_inicial is not None
         assert pmo.volume_armazenado_inicial is not None
         assert pmo.penalidade_violacao_outros_usos is not None
@@ -347,6 +401,8 @@ def test_atributos_encontrados_pmo():
         assert pmo.penalidade_violacao_evaporacao is not None
         assert pmo.penalidade_violacao_turbinamento_maximo is not None
         assert pmo.penalidade_violacao_turbinamento_minimo is not None
+        assert pmo.geracao_minima_usinas_termicas is not None
+        assert pmo.geracao_maxima_usinas_termicas is not None
 
 
 def test_atributos_nao_encontrados_pmo():
@@ -358,6 +414,7 @@ def test_atributos_nao_encontrados_pmo():
         assert pmo.custo_operacao_total is None
         assert pmo.desvio_custo_operacao_total is None
         assert pmo.produtibilidades_equivalentes is None
+        assert pmo.energia_armazenada_maxima is None
         assert pmo.energia_armazenada_inicial is None
         assert pmo.volume_armazenado_inicial is None
         assert pmo.penalidade_violacao_outros_usos is None
@@ -367,6 +424,8 @@ def test_atributos_nao_encontrados_pmo():
         assert pmo.penalidade_violacao_evaporacao is None
         assert pmo.penalidade_violacao_turbinamento_maximo is None
         assert pmo.penalidade_violacao_turbinamento_minimo is None
+        assert pmo.geracao_minima_usinas_termicas is None
+        assert pmo.geracao_maxima_usinas_termicas is None
 
 
 def test_eq_pmo():

tests/newave/test_vazinat.py

@@ -0,0 +1,66 @@
+from inewave.newave.modelos.vazinat import SecaoDadosVazinat
+from inewave.newave.vazinat import Vazinat
+from inewave.config import MAX_ANOS_HISTORICO
+
+from tests.mocks.mock_open import mock_open
+from unittest.mock import MagicMock, patch
+
+
+ARQ_TESTE = "./tests/mocks/arquivos/vazinat.dat"
+
+NUM_CONFIGURACOES = 1
+NUM_UHES = 164
+ANO_INICIO_HISTORICO = 1931
+NUM_ENTRADAS = NUM_CONFIGURACOES * NUM_UHES * 12 * MAX_ANOS_HISTORICO
+
+
+def test_secao_vazinat():
+    r = SecaoDadosVazinat()
+    with open(ARQ_TESTE, "rb") as fp:
+        r.read(
+            fp,
+            numero_uhes=NUM_UHES,
+            ano_inicio_historico=ANO_INICIO_HISTORICO,
+            numero_configuracoes=NUM_CONFIGURACOES,
+        )
+
+    assert len(r.data) == NUM_ENTRADAS
+
+
+def test_atributos_encontrados_vazinat():
+    h = Vazinat.read(
+        ARQ_TESTE,
+        numero_uhes=NUM_UHES,
+        ano_inicio_historico=ANO_INICIO_HISTORICO,
+        numero_configuracoes=NUM_CONFIGURACOES,
+    )
+    assert h.series is not None
+    assert h.series.isna().sum().sum() == 0
+
+
+def test_atributos_nao_encontrados_vazinat():
+    m: MagicMock = mock_open(read_data="")
+    with patch("builtins.open", m):
+        h = Vazinat.read(
+            ARQ_TESTE,
+            numero_uhes=NUM_UHES,
+            ano_inicio_historico=ANO_INICIO_HISTORICO,
+            numero_configuracoes=NUM_CONFIGURACOES,
+        )
+        assert h.series.isna().sum().sum() == NUM_ENTRADAS
+
+
+def test_eq_vazinat():
+    h1 = Vazinat.read(
+        ARQ_TESTE,
+        numero_uhes=NUM_UHES,
+        ano_inicio_historico=ANO_INICIO_HISTORICO,
+        numero_configuracoes=NUM_CONFIGURACOES,
+    )
+    h2 = Vazinat.read(
+        ARQ_TESTE,
+        numero_uhes=NUM_UHES,
+        ano_inicio_historico=ANO_INICIO_HISTORICO,
+        numero_configuracoes=NUM_CONFIGURACOES,
+    )
+    assert h1 == h2

tests/nwlistop/test_mediasmerc.py

@@ -3,15 +3,11 @@ from inewave.nwlistop.mediasmerc import Mediasmerc
 import pandas as pd  # type: ignore
 
 
-leitor = Mediasmerc.le_arquivo("tests/_arquivos")
+ARQ_TESTE = "tests/_arquivos/MEDIAS-MERC.CSV"
 
 
 def test_eq_mediasmerc():
-    leitor2 = Mediasmerc.le_arquivo("tests/_arquivos")
+    leitor = Mediasmerc.read(ARQ_TESTE)
+    leitor2 = Mediasmerc.read(ARQ_TESTE)
     assert leitor == leitor2
-
-
-def test_neq_mediasmerc():
-    leitor2 = Mediasmerc.le_arquivo("tests/_arquivos")
-    leitor2.medias = pd.DataFrame()
-    assert leitor != leitor2
+    assert leitor.valores.shape == (228, 54)

tests/nwlistop/test_mediasree.py

@@ -0,0 +1,12 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-REE.CSV do NWLISTOP
+from inewave.nwlistop.mediasree import Mediasree
+
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-REE.CSV"
+
+
+def test_eq_mediasree():
+    leitor = Mediasree.read(ARQ_TESTE)
+    leitor2 = Mediasree.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (720, 49)

tests/nwlistop/test_mediasrep.py

@@ -0,0 +1,12 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-REP.CSV do NWLISTOP
+from inewave.nwlistop.mediasrep import Mediasrep
+
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-REP.CSV"
+
+
+def test_eq_mediasrep():
+    leitor = Mediasrep.read(ARQ_TESTE)
+    leitor2 = Mediasrep.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (0, 2)

tests/nwlistop/test_mediasrhq.py

@@ -0,0 +1,12 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-RHQ.CSV do NWLISTOP
+from inewave.nwlistop.mediasrhq import Mediasrhq
+
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-RHQ.CSV"
+
+
+def test_eq_mediasrhq():
+    leitor = Mediasrhq.read(ARQ_TESTE)
+    leitor2 = Mediasrhq.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (0, 2)

tests/nwlistop/test_mediasrhv.py

@@ -0,0 +1,11 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-RHV.CSV do NWLISTOP
+from inewave.nwlistop.mediasrhv import Mediasrhv
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-RHV.CSV"
+
+
+def test_eq_mediasrhv():
+    leitor = Mediasrhv.read(ARQ_TESTE)
+    leitor2 = Mediasrhv.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (0, 2)

tests/nwlistop/test_mediassin.py

@@ -1,17 +1,12 @@
 # Rotinas de testes associadas ao arquivo MEDIAS-SIN.CSV do NWLISTOP
 from inewave.nwlistop.mediassin import Mediassin
-import pandas as pd  # type: ignore
 
 
-leitor = Mediassin.le_arquivo("tests/_arquivos")
+ARQ_TESTE = "tests/_arquivos/MEDIAS-SIN.CSV"
 
 
 def test_eq_mediassin():
-    leitor2 = Mediassin.le_arquivo("tests/_arquivos")
+    leitor = Mediassin.read(ARQ_TESTE)
+    leitor2 = Mediassin.read(ARQ_TESTE)
     assert leitor == leitor2
-
-
-def test_neq_mediassin():
-    leitor2 = Mediassin.le_arquivo("tests/_arquivos")
-    leitor2.medias = pd.DataFrame()
-    assert leitor != leitor2
+    assert leitor.valores.shape == (57, 55)

tests/nwlistop/test_mediasusie.py

@@ -0,0 +1,12 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-USIE.CSV do NWLISTOP
+from inewave.nwlistop.mediasusie import Mediasusie
+
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-USIE.CSV"
+
+
+def test_eq_mediasusie():
+    leitor = Mediasusie.read(ARQ_TESTE)
+    leitor2 = Mediasusie.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (0, 2)

tests/nwlistop/test_mediasusih.py

@@ -0,0 +1,11 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-UISH.CSV do NWLISTOP
+from inewave.nwlistop.mediasusih import Mediasusih
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-USIH.CSV"
+
+
+def test_eq_mediasusih():
+    leitor = Mediasusih.read(ARQ_TESTE)
+    leitor2 = Mediasusih.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (9840, 30)

tests/nwlistop/test_mediasusit.py

@@ -0,0 +1,12 @@
+# Rotinas de testes associadas ao arquivo MEDIAS-USIT.CSV do NWLISTOP
+from inewave.nwlistop.mediasusit import Mediasusit
+
+
+ARQ_TESTE = "tests/_arquivos/MEDIAS-USIT.CSV"
+
+
+def test_eq_mediasusit():
+    leitor = Mediasusit.read(ARQ_TESTE)
+    leitor2 = Mediasusit.read(ARQ_TESTE)
+    assert leitor == leitor2
+    assert leitor.valores.shape == (7680, 5)

inewave/_utils/leituracsv.py

@@ -1,57 +0,0 @@
-import os
-from abc import abstractmethod
-from traceback import print_exc
-import pandas as pd  # type: ignore
-
-
-class LeituraCSV:
-    """
-    Classe com utilidades gerais para leitura de arquivos
-    .csv do NEWAVE.
-    """
-
-    def __init__(self, diretorio: str):
-        self._diretorio = diretorio
-        self._dados = pd.DataFrame()
-
-    def _le_tabela_arquivo(self, caminho: str):
-        """
-        Faz a leitura dos blocos de dados do arquivo.
-        """
-        self._dados = pd.read_csv(
-            caminho, sep=",", encoding="utf-8", index_col=1
-        )
-
-    def _le_arquivo_em_diretorio(
-        self, diretorio: str, nome_arquivo: str
-    ) -> None:
-        """
-        Faz a leitura do arquivo em um diretorio.
-        """
-        try:
-            caminho = os.path.join(diretorio, nome_arquivo)
-            self._le_tabela_arquivo(caminho)
-            self.processa_dados_lidos()
-        except Exception:
-            print_exc()
-
-    def le_arquivo(self, nome_arquivo: str) -> pd.DataFrame:
-        """
-        Método para ler um arquivo e retornar o objeto
-        devido da classe em particular.
-        """
-        self._le_arquivo_em_diretorio(self._diretorio, nome_arquivo)
-        return self._dados
-
-    @abstractmethod
-    def processa_dados_lidos(self):
-        """ """
-        pass
-
-    @property
-    def dados(self) -> pd.DataFrame:
-        """
-        Dados de leitura obtidos pelo blocos, organizados
-        para realizar a criação do objeto associado.
-        """
-        return self._dados