Alcance de fixtures

parent ad4667dd
import networkx as nx import networkx as nx
import pandas as pd import pandas as pd
import pytest import pytest
from redsem import main from redsem import main
@pytest.fixture() @pytest.fixture(scope='module')
def df_prueba(): def df_prueba():
datos = {'palabras': ['prueba', 'para', 'redes', 'semánticas'], datos = {'palabras': ['prueba', 'para', 'redes', 'semánticas'],
'posicion': [1, 2, 3, 4]} 'posicion': [1, 2, 3, 4]}
return pd.DataFrame(datos) return pd.DataFrame(datos)
@pytest.fixture() @pytest.fixture(scope='module')
def red_prueba(): def red_prueba():
palabras_estimulo = ['covid_semilla', 'obesidad_semilla'] palabras_estimulo = ['covid_semilla', 'obesidad_semilla']
palabras_respuesta = ['salud', 'ansiedad', 'higiene', 'comida'] palabras_respuesta = ['salud', 'ansiedad', 'higiene', 'comida']
...@@ -34,4 +35,4 @@ def red_prueba(): ...@@ -34,4 +35,4 @@ def red_prueba():
@pytest.mark.parametrize("ultima", [1, 2, 3, 4]) @pytest.mark.parametrize("ultima", [1, 2, 3, 4])
def test_filtrado_correcto(df_prueba, ultima): def test_filtrado_correcto(df_prueba, ultima):
df_filtrado = main.filtrar_primeras_posiciones(df_prueba, ultima_posicion=ultima) df_filtrado = main.filtrar_primeras_posiciones(df_prueba, ultima_posicion=ultima)
assert len(df_filtrado) == ultima assert df_filtrado['posicion'].max() <= ultima
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment