Skip to content

C
ConceptExtractor
Switch branch/tag
Find file Normal viewHistoryPermalink
Dates_PruebasUnitarias.py 6.82 KB
Newer Older
Lucia Catalan Gris's avatar
Upload New File  
Lucia Catalan Gris committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 
# -*- coding: utf-8 -*-
"""
Created on Fri Feb 12 17:24:53 2021

@author: Lucia
"""

from Dates import fechas_tablas
import pandas.testing as pdt
import pandas as pd

print("---CASOS FECHAS---")

tabla_documentos = pd.DataFrame({'ID': [8,9], 'EHR': [2,3]})

resultado_date = pd.DataFrame({'date_id':[0,1,2,3,4,5,6],
                               'note_date': ["2021-07-23","2021-07-23","2021-07-23","2021-07-23","2021-07-23","2021-07-23","2021-07-23"]})
resultado_note_date = pd.DataFrame({'note_id': [8,8,8,8,8,8,8],'date_id':[0,1,2,3,4,5,6],
                                    'begin': [0,0,0,0,0,0,0],'end':[10,10,10,10,10,10,10]})

print("TEST 1 -> fechas con -")
anotaciones = [[{"word": "23 - 7 - 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23-7-2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23-7se-2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 - 7 - 2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23-7-2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23-7 - 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 - 7-2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10}]]

fechas = fechas_tablas(anotaciones, tabla_documentos)
try:
    pdt.assert_frame_equal(fechas[0], resultado_date)
    pdt.assert_frame_equal(fechas[1], resultado_note_date)
    print('OK')
except AssertionError:
    print('TEST 1 NOT PASSED')

print("TEST 2 -> fechas con /")
anotaciones = [[{"word": "23 / 7 / 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                      {"word": "23/7/2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                      {"word": "23/7se/2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                      {"word": "23 / 7 / 2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                      {"word": "23/7/2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                      {"word": "23/7 / 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                      {"word": "23 / 7/2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10}]]

fechas = fechas_tablas(anotaciones, tabla_documentos)
try:
    pdt.assert_frame_equal(fechas[0], resultado_date)
    pdt.assert_frame_equal(fechas[1], resultado_note_date)
    print('OK')
except AssertionError:
    print('TEST 2 NOT PASSED')
    
print("TEST 3 -> fechas con .")
anotaciones = [[{"word": "23 . 7 . 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23.7.2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23.7se.2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 . 7 . 2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23.7.2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23.7 . 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 . 7.2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10}]]

fechas = fechas_tablas(anotaciones, tabla_documentos)
try:
    pdt.assert_frame_equal(fechas[0], resultado_date)
    pdt.assert_frame_equal(fechas[1], resultado_note_date)
    print('OK')
except AssertionError:
    print('TEST 3 NOT PASSED')
    
print("TEST 4 -> fechas con letra")
anotaciones = [[{"word": "23 de julio del 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 julio 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23se julio 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 de julio 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 de julio 2021se","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 / julio / 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23.julio.2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10}]]


fechas = fechas_tablas(anotaciones, tabla_documentos)
try:
    pdt.assert_frame_equal(fechas[0], resultado_date)
    pdt.assert_frame_equal(fechas[1], resultado_note_date)
    print('OK')
except AssertionError:
    print('TEST 4 NOT PASSED')

print("---CASO INPUT VACIO---")

print("TEST 5 -> anotaciones vacias")
anotaciones_vacia = [[]]
tabla_documentos_vacia = pd.DataFrame({'ID': [8,9], 'EHR': [2,3]})
fechas = fechas_tablas(anotaciones_vacia, tabla_documentos_vacia)

if (fechas[0].empty) and (fechas[1].empty):
    print('OK')
else:
    print('TEST 5 NOT PASSED')

print("---CASO INPUTS INCORRECTOS---")

print("TEST 6 -> no es una fecha")
anotaciones = [[{"word": "hola","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "32 julio 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "2435271","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 / 13 / 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "no es una fecha","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "no","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "fecha","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10}]]

fechas = fechas_tablas(anotaciones, tabla_documentos)

if (fechas[0].empty) and (fechas[1].empty):
    print('OK')
else:
    print('TEST 6 NOT PASSED')

print("TEST 7 -> inputs al reves")
'''
anotaciones = [[{"word": "hola","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "32 julio 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "2435271","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "23 / 13 / 2021","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "no es una fecha","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": "","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10},
                {"word": ".","score": 0,"entity": "B_DATE","index": 0,"start": 0,"end": 10}]]

fechas = fechas_tablas(tabla_documentos, anotaciones)
try:
    pdt.assert_frame_equal(fechas[0], resultado_date_vacia)
    pdt.assert_frame_equal(fechas[1], resultado_note_date_vacia)
    print('OK')
except AssertionError:
    print('TEST 7 NOT PASSED')
    '''