#!/usr/bin/python

# Basically, add '../src' to our path.
import sys, os, site
site.addsitedir(os.path.dirname(os.path.abspath(sys.argv[0])) + '/../src')

import unittest
import Census
from Tests.Unit import CensusTest
from Tests.Unit import DistributionTest
from Tests.Unit import FileUtilsTest
from Tests.Unit import GeometryTest
from Tests.Unit import KMLTest
from Tests.Unit import LEHDTest
from Tests.Unit import SimplexMethodTest
from Tests.Unit import SummaryFile1Test
from Tests.Unit import StringUtilsTest
from Tests.Unit import UniformDistributionTest

suite = unittest.TestSuite()
suite.addTest(CensusTest.suite())
suite.addTest(FileUtilsTest.suite())
suite.addTest(GeometryTest.suite())
suite.addTest(KMLTest.suite())
suite.addTest(SummaryFile1Test.suite())
suite.addTest(StringUtilsTest.suite())
suite.addTest(LEHDTest.suite())
suite.addTest(UniformDistributionTest.suite())
suite.addTest(DistributionTest.suite())
suite.addTest(SimplexMethodTest.suite())
unittest.TextTestRunner(verbosity=2).run(suite)

try:
    # The Census.Database initialization will raise an exception
    # if the database cannot be accessed.
    cdb = Census.Database()
    db_suite = unittest.TestSuite()
    db_suite.addTest(CensusTest.db_suite())
    unittest.TextTestRunner(verbosity=2).run(db_suite)
except:
    pass