378 lines
15 KiB
Python
378 lines
15 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2007 Donald N. Allingham
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
|
|
# $Id$
|
|
|
|
""" Unittest for testing dates """
|
|
|
|
import unittest
|
|
import os
|
|
import sys
|
|
import traceback
|
|
import locale
|
|
import gettext
|
|
|
|
if "GRAMPSI18N" in os.environ:
|
|
loc = os.environ["GRAMPSI18N"]
|
|
else:
|
|
loc = "/usr/share/locale"
|
|
|
|
try:
|
|
locale.setlocale(locale.LC_ALL,'C')
|
|
locale.setlocale(locale.LC_ALL,'')
|
|
except locale.Error:
|
|
pass
|
|
except ValueError:
|
|
pass
|
|
|
|
from test import test_util
|
|
test_util.path_append_parent()
|
|
|
|
import Config
|
|
import DateHandler
|
|
from DateHandler import parser as _dp
|
|
from DateHandler import displayer as _dd
|
|
from gen.lib.date import Date, Span
|
|
|
|
gettext.textdomain("gramps")
|
|
gettext.install("gramps",loc,unicode=1)
|
|
|
|
date_tests = {}
|
|
|
|
# first the "basics".
|
|
testset = "basic test"
|
|
dates = []
|
|
calendar = Date.CAL_GREGORIAN
|
|
for quality in (Date.QUAL_NONE, Date.QUAL_ESTIMATED, Date.QUAL_CALCULATED):
|
|
for modifier in (Date.MOD_NONE, Date.MOD_BEFORE, Date.MOD_AFTER, Date.MOD_ABOUT):
|
|
for month in range(1,13):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,month,1789,False),"Text comment")
|
|
dates.append( d)
|
|
for modifier in (Date.MOD_RANGE, Date.MOD_SPAN):
|
|
for month1 in range(1,13):
|
|
for month2 in range(1,13):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,month1,1789,False,5,month2,1876,False),"Text comment")
|
|
dates.append( d)
|
|
modifier = Date.MOD_TEXTONLY
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,Date.EMPTY,"This is a textual date")
|
|
dates.append( d)
|
|
date_tests[testset] = dates
|
|
|
|
# incomplete dates (day or month missing)
|
|
testset = "partial date"
|
|
dates = []
|
|
calendar = Date.CAL_GREGORIAN
|
|
for quality in (Date.QUAL_NONE, Date.QUAL_ESTIMATED, Date.QUAL_CALCULATED):
|
|
for modifier in (Date.MOD_NONE, Date.MOD_BEFORE, Date.MOD_AFTER, Date.MOD_ABOUT):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,11,1789,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,0,1789,False),"Text comment")
|
|
dates.append( d)
|
|
for modifier in (Date.MOD_RANGE, Date.MOD_SPAN):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,10,1789,False,0,11,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,10,1789,False,0,0,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,10,1789,False,5,11,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,10,1789,False,0,11,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,10,1789,False,0,0,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,0,1789,False,5,11,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,0,1789,False,0,11,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(0,0,1789,False,0,0,1876,False),"Text comment")
|
|
dates.append( d)
|
|
date_tests[testset] = dates
|
|
|
|
# slash-dates
|
|
testset = "slash-dates"
|
|
dates = []
|
|
calendar = Date.CAL_GREGORIAN
|
|
for quality in (Date.QUAL_NONE, Date.QUAL_ESTIMATED, Date.QUAL_CALCULATED):
|
|
for modifier in (Date.MOD_NONE, Date.MOD_BEFORE, Date.MOD_AFTER, Date.MOD_ABOUT):
|
|
# normal date
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,11,1789,True),"Text comment")
|
|
dates.append( d)
|
|
for modifier in (Date.MOD_RANGE, Date.MOD_SPAN):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,11,1789,True,5,10,1876,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,11,1789,False,5,10,1876,True),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,11,1789,True,5,10,1876,True),"Text comment")
|
|
dates.append( d)
|
|
date_tests[testset] = dates
|
|
|
|
# BCE
|
|
testset = "B. C. E."
|
|
dates = []
|
|
calendar = Date.CAL_GREGORIAN
|
|
for quality in (Date.QUAL_NONE, Date.QUAL_ESTIMATED, Date.QUAL_CALCULATED):
|
|
for modifier in (Date.MOD_NONE, Date.MOD_BEFORE, Date.MOD_AFTER, Date.MOD_ABOUT):
|
|
# normal date
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,11,-90,False),"Text comment")
|
|
dates.append( d)
|
|
for modifier in (Date.MOD_RANGE, Date.MOD_SPAN):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(5,10,-90,False,4,11,-90,False),"Text comment")
|
|
dates.append( d)
|
|
d = Date()
|
|
date_tests[testset] = dates
|
|
|
|
# test for all other different calendars
|
|
testset = "Non-gregorian"
|
|
dates = []
|
|
for calendar in (Date.CAL_JULIAN, Date.CAL_HEBREW, Date.CAL_ISLAMIC, Date.CAL_FRENCH, Date.CAL_PERSIAN, Date.CAL_SWEDISH):
|
|
for quality in (Date.QUAL_NONE, Date.QUAL_ESTIMATED, Date.QUAL_CALCULATED):
|
|
for modifier in (Date.MOD_NONE, Date.MOD_BEFORE, Date.MOD_AFTER, Date.MOD_ABOUT):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,11,1789,False),"Text comment")
|
|
dates.append( d)
|
|
for modifier in (Date.MOD_RANGE, Date.MOD_SPAN):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,10,1789,False,5,11,1876,False),"Text comment")
|
|
dates.append( d)
|
|
quality = Date.QUAL_NONE
|
|
modifier = Date.MOD_NONE
|
|
for calendar in (Date.CAL_JULIAN, Date.CAL_ISLAMIC, Date.CAL_PERSIAN, Date.CAL_SWEDISH):
|
|
for month in range(1,13):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,month,1789,False),"Text comment")
|
|
dates.append( d)
|
|
for calendar in (Date.CAL_HEBREW, Date.CAL_FRENCH):
|
|
for month in range(1,14):
|
|
d = Date()
|
|
d.set(quality,modifier,calendar,(4,month,1789,False),"Text comment")
|
|
dates.append( d)
|
|
date_tests[testset] = dates
|
|
|
|
# now run the tests using all available date formats
|
|
cal_str = [ "CAL_GREGORIAN", "CAL_JULIAN", "CAL_HEBREW", "CAL_FRENCH", "CAL_PERSIAN", "CAL_ISLAMIC", "CAL_SWEDISH"]
|
|
mod_str = ["MOD_NONE", "MOD_BEFORE", "MOD_AFTER", "MOD_ABOUT", "MOD_RANGE", "MOD_SPAN", "MOD_TEXTONLY"]
|
|
qua_str = ["QUAL_NONE", "QUAL_ESTIMATED", "QUAL_CALCULATED"]
|
|
formats = DateHandler.get_date_formats()
|
|
|
|
class Eval(unittest.TestCase):
|
|
def __init__(self, method_name, dateval, e1, e2, e3):
|
|
self.__dict__[method_name] = lambda: eval_func(dateval, e1, e2, e3)
|
|
unittest.TestCase.__init__(self, method_name)
|
|
|
|
def eval_func(dateval, e1, e2, e3):
|
|
exec(e1)
|
|
exec(e2)
|
|
exec(e3)
|
|
#print datestr, ndate, ntxt
|
|
assert(dateval.is_equal(ndate), "dates are not the same")
|
|
|
|
def suite3():
|
|
suite = unittest.TestSuite()
|
|
count = 1
|
|
for testset in date_tests:
|
|
for format in range( len( DateHandler.get_date_formats())):
|
|
DateHandler.set_format(format)
|
|
for dateval in date_tests[testset]:
|
|
if dateval.modifier != Date.MOD_TEXTONLY:
|
|
dateval.text = "Comment. Format: %s" % DateHandler.get_date_formats()[format]
|
|
suite.addTest(Eval("test_eval%04d" % count, dateval,
|
|
"datestr = _dd.display(dateval)",
|
|
"ndate = _dp.parse(datestr)",
|
|
"ntxt = _dd.display(ndate)"))
|
|
count += 1
|
|
return suite
|
|
|
|
class Tester(unittest.TestCase):
|
|
def __init__(self, method_name, part, testdata):
|
|
self.__dict__[method_name + ("-%d" % part)] = lambda: self.helper(part, *testdata)
|
|
unittest.TestCase.__init__(self, method_name + ("-%d" % part))
|
|
|
|
def helper(self, part, d1, d2, expected1, expected2 = None):
|
|
"""
|
|
Tests two GRAMPS dates to see if they match.
|
|
"""
|
|
if expected2 is None:
|
|
expected2 = expected1
|
|
date1 = _dp.parse(d1)
|
|
date2 = _dp.parse(d2)
|
|
if part == 1:
|
|
val = date2.match(date1)
|
|
self.assertTrue(val == expected1, "'%s' and '%s' did not match" % (d1, d2))
|
|
else:
|
|
val = date1.match(date2)
|
|
self.assertTrue(val == expected2, "'%s' and '%s' did not match" % (d2, d1))
|
|
|
|
def suite():
|
|
""" interface to automated test runner test/regrtest.py """
|
|
Config.set(Config.DATE_BEFORE_RANGE, 9999)
|
|
Config.set(Config.DATE_AFTER_RANGE, 9999)
|
|
Config.set(Config.DATE_ABOUT_RANGE, 10)
|
|
# most are symmetric: #date1, date2, does d1 match d2? does d2 match d1?
|
|
tests = [("before 1960", "before 1961", True),
|
|
("before 1960", "before 1960", True),
|
|
("before 1961", "before 1961", True),
|
|
("jan 1, 1960", "jan 1, 1960", True),
|
|
("dec 31, 1959", "dec 31, 1959", True),
|
|
("before 1960", "jan 1, 1960", False),
|
|
("before 1960", "dec 31, 1959", True),
|
|
("abt 1960", "1960", True),
|
|
("abt 1960", "before 1960", True),
|
|
("1960", "1960", True),
|
|
("1960", "after 1960", False),
|
|
("1960", "before 1960", False),
|
|
("abt 1960", "abt 1960", True),
|
|
("before 1960", "after 1960", False),
|
|
("after jan 1, 1900", "jan 2, 1900", True),
|
|
("abt jan 1, 1900", "jan 1, 1900", True),
|
|
("from 1950 to 1955", "1950", True),
|
|
("from 1950 to 1955", "1951", True),
|
|
("from 1950 to 1955", "1952", True),
|
|
("from 1950 to 1955", "1953", True),
|
|
("from 1950 to 1955", "1954", True),
|
|
("from 1950 to 1955", "1955", True),
|
|
("from 1950 to 1955", "1956", False),
|
|
("from 1950 to 1955", "dec 31, 1955", True),
|
|
("from 1950 to 1955", "jan 1, 1955", True),
|
|
("from 1950 to 1955", "dec 31, 1949", False),
|
|
("from 1950 to 1955", "jan 1, 1956", False),
|
|
("after jul 4, 1980", "jul 4, 1980", False),
|
|
("after jul 4, 1980", "before jul 4, 1980", False),
|
|
("after jul 4, 1980", "about jul 4, 1980", True),
|
|
("after jul 4, 1980", "after jul 4, 1980", True),
|
|
("between 1750 and 1752", "1750", True),
|
|
("between 1750 and 1752", "about 1750", True),
|
|
("between 1750 and 1752", "between 1749 and 1750", True),
|
|
("between 1750 and 1752", "1749", False),
|
|
("invalid date", "invalid date", True),
|
|
("invalid date", "invalid", False, True),
|
|
("invalid date 1", "invalid date 2", False),
|
|
("abt jan 1, 2000", "dec 31, 1999", True),
|
|
("jan 1, 2000", "dec 31, 1999", False),
|
|
("aft jan 1, 2000", "dec 31, 1999", False),
|
|
("after jan 1, 2000", "after dec 31, 1999", True),
|
|
("after dec 31, 1999", "after jan 1, 2000", True),
|
|
("1 31, 2000", "jan 1, 2000", False),
|
|
("dec 31, 1999", "jan 1, 2000", False),
|
|
("jan 1, 2000", "before dec 31, 1999", False),
|
|
("aft jan 1, 2000", "before dec 31, 1999", False),
|
|
("before jan 1, 2000", "after dec 31, 1999", False),
|
|
("jan 1, 2000/1", "jan 1, 2000", False),
|
|
("jan 1, 2000/1", "jan 1, 2001", False),
|
|
("jan 1, 2000/1", "jan 1, 2000/1", True),
|
|
("jan 1, 2000/1", "jan 14, 2001", True),
|
|
("jan 1, 2000/1", "jan 1, 2001 (julian)", True),
|
|
("about 1984", "about 2005", False),
|
|
("about 1990", "about 2005", True),
|
|
("about 2007", "about 2006", True),
|
|
("about 1995", "after 2000", True),
|
|
("about 1995", "after 2005", False),
|
|
("about 2007", "about 2003", True),
|
|
("before 2007", "2000", True),
|
|
# offsets
|
|
# different calendar, same date
|
|
("1800-8-3", "15 Thermidor 8 (French Republican)", True),
|
|
("after 1800-8-3", "before 15 Thermidor 8 (French Republican)", False),
|
|
("ab cd", "54 ab cd 2000", True, False),
|
|
("1700-02-29 (Julian)", "1700-03-01 (Swedish)", True),
|
|
("1706-12-31 (Julian)", "1707-01-01 (Swedish)", True),
|
|
("1712-02-28 (Julian)", "1712-02-29 (Swedish)", True),
|
|
("1712-02-29 (Julian)", "1712-02-30 (Swedish)", True),
|
|
]
|
|
suite = unittest.TestSuite()
|
|
count = 1
|
|
for test in tests:
|
|
suite.addTest(Tester('test_match%04d' % count, 1, test))
|
|
suite.addTest(Tester('test_match%04d' % count, 2, test))
|
|
count += 1
|
|
return suite
|
|
|
|
def assert_func(exp1, exp2):
|
|
e1 = eval(exp1)
|
|
e2 = eval(exp2)
|
|
assert e1 == e2, "%s should be %s but was %s" % (exp1, e2, e1)
|
|
|
|
class Assert(unittest.TestCase):
|
|
def __init__(self, method_name, part, exp1, exp2):
|
|
self.__dict__[method_name + ("-%d" % part)] = \
|
|
lambda: assert_func(exp1, exp2)
|
|
unittest.TestCase.__init__(self, method_name + ("-%d" % part))
|
|
|
|
def suite2():
|
|
""" interface to automated test runner test/regrtest.py """
|
|
Config.set(Config.DATE_BEFORE_RANGE, 9999)
|
|
Config.set(Config.DATE_AFTER_RANGE, 9999)
|
|
Config.set(Config.DATE_ABOUT_RANGE, 10)
|
|
tests = [
|
|
# Date +/- int/tuple -> Date
|
|
("Date(2008, 1, 1) - 1", "Date(2007, 1, 1)"),
|
|
("Date(2008, 1, 1) + 1", "Date(2009, 1, 1)"),
|
|
("Date(2008, 1, 1) - (0,0,1)", "Date(2007, 12, 31)"),
|
|
("Date(2008, 1, 1) - (0,0,2)", "Date(2007, 12, 30)"),
|
|
("Date(2008) - (0,0,1)", "Date(2007, 12, 31)"),
|
|
("Date(2008) - 1", "Date(2007, 1, 1)"),
|
|
("Date(2008, 12, 31) + (0, 0, 1)", "Date(2009, 1, 1)"),
|
|
("Date(2000,1,1) - (0,11,0)", "Date(1999,02,01)"),
|
|
("Date(2000,1,1) - (0,1,0)", "Date(1999, 12, 1)"),
|
|
("Date(2008, 1, 1) + (0, 0, 32)", "Date(2008, 02, 02)"),
|
|
("Date(2008, 2, 1) + (0, 0, 32)", "Date(2008, 03, 04)"),
|
|
("Date(2000) - (0, 1, 0)", "Date(1999, 12, 1)"),
|
|
("Date(2000) + (0, 1, 0)", "Date(2000, 1, 0)"), # Ok?
|
|
("Date(2000, 1, 1) - (0, 1, 0)", "Date(1999, 12, 1)"),
|
|
("Date(2000, 1, 1) - 1", "Date(1999, 1, 1)"),
|
|
("Date(2000) - 1", "Date(1999)"),
|
|
("Date(2000) + 1", "Date(2001)"),
|
|
# Date +/- Date -> Span
|
|
("(Date(1876,5,7) - Date(1876,5,1)).tuple()", "(0, 0, 6)"),
|
|
("(Date(1876,5,7) - Date(1876,4,30)).tuple()", "(0, 0, 7)"),
|
|
("(Date(2000,1,1) - Date(1999,2,1)).tuple()", "(0, 11, 0)"),
|
|
("(Date(2000,1,1) - Date(1999,12,1)).tuple()", "(0, 1, 0)"),
|
|
("(Date(2007, 12, 23) - Date(1963, 12, 4)).tuple()", "(44, 0, 19)"),
|
|
("(Date(1963, 12, 4) - Date(2007, 12, 23)).tuple()", "(-44, 0, -19)"),
|
|
]
|
|
suite = unittest.TestSuite()
|
|
count = 1
|
|
for (exp1, exp2) in tests:
|
|
suite.addTest(Assert('test_assert%04d' % count, 1, exp1, exp2))
|
|
count += 1
|
|
return suite
|
|
|
|
if __name__ == "__main__":
|
|
unittest.TextTestRunner().run(suite3())
|
|
unittest.TextTestRunner().run(suite2())
|
|
unittest.TextTestRunner().run(suite())
|