def test_valida_dificultad_facil(self):
     # Arrange
     p = Palabra('agua')
     esperado = 'FACIL'
     # Act
     resultado = p.getDificultad()
     # Assert
     assert esperado == resultado  
 def test_valida_dificultad_media(self):
     # Arrange
     p = Palabra('aprender')
     esperado = 'MEDIA'
     # Act
     resultado = p.getDificultad()
     # Assert
     assert esperado == resultado
 def test_valida_arriesgar_palabra_correcta(self):
     # Arrange
     p = Palabra('casa')
     esperado = True
     # Act
     resultado = p.validaPalabra('casa')
     # Assert
     assert esperado == resultado 
 def test_valida_arriesgar_palabra_incorrecta(self):
     # Arrange
     p = Palabra('casa')
     esperado = False
     # Act
     resultado = p.validaPalabra('helicoptero')
     # Assert
     assert esperado == resultado
 def test_valida_letra_a_en_palabra_casa(self):
     # Arrange
     p = Palabra('casa')
     esperado = True
     # Act
     resultado = p.validaLetra('A') 
     # Assert
     assert esperado == resultado
 def test_valida_estado_palabra_casa(self):
     # Arrange
     p = Palabra('casa')
     esperado = '****'
     # Act
     resultado = p.getEstado() 
     # Assert
     assert esperado == resultado
 def test_valida_dificultad_dificil(self):
     # Arrange
     p = Palabra('dificultad')
     esperado = 'DIFICIL'
     # Act
     resultado = p.getDificultad()
     # Assert
     assert esperado == resultado
 def test_valida_estado_palabra_casa_con_letra_A(self):
     # Arrange
     p = Palabra('casa')
     esperado = '*A*A'
     # Act
     p.validaLetra('A')
     resultado = p.getEstado() 
     # Assert
     assert esperado == resultado
 def test_valida_agregar_letra_a_lista(self):
     # Arrange
     p = Palabra('casa')
     esperado = ['A', 'S']
     # Act
     p.validaLetra('A')
     p.validaLetra('S')
     resultado = p.getLetrasArriesgadas() 
     # Assert
     assert esperado == resultado
import pymongo
from clases import Palabra

f = open("words.txt", 'r', errors='ignore')
words = f.readlines()
new_words = []
for word in words:
    new_words.append(Palabra(word.rstrip("\n")))

client = pymongo.MongoClient(
    "mongodb+srv://test:[email protected]/<dbname>?retryWrites=true&w=majority"
)

mydb = client["Ahorcado"]
mycol = mydb["Palabras"]

for word in new_words:
    mydict = {
        "palabra": word.getPalabra(),
        "dificultad": word.dificultad,
        "puntajeDificultad": word.puntajeDificultad
    }
    x = mycol.insert_one(mydict)