def menu(): entra = True while entra: print("1.-Crear BD\n2.-Restaurar BD\n3.-Salir") numero = None try: numero = int(input("Introduce una opcion: ")) except Exception as e: numero = -1 if numero == 1: atributos = input( "Escribir los atributos de la BD separados por comas(,):") listaAtributos = atributos.replace(" ", "").split(",") bienConstruidos = True for atri in listaAtributos: if atri == "": bienConstruidos = False if bienConstruidos: base = BaseDeDatos() base.crearBD(atributos) menuSecundario(base) else: print("Los atributos estan mal construidos") elif numero == 2: nomBD = input("Nombre del archivo donde se encuentra la BD:") f = open(nomBD, "r") atributosTxt = f.read() base = BaseDeDatos() base.crearBDTxt(atributosTxt) f.close() menuSecundario(base) elif numero == 3: print("Saliendo...") exit()
def test_creaBDTxt(self): base = BaseDeDatos() self.assertEqual(base.crearBDTxt("hermanos"), "Se creo la BD")