import unittest import datetime import pytz import sys sys.path.append("..") from dre.decision import * import dre.actions as actions from database import database run = database.get_default_values_conf("tests")['general_config']["run"] class RunTest(unittest.TestCase): def testRunAction(self): aRunAction = actions.GaussDistFromIdeal(datetime.datetime.now(tz=pytz.UTC), Loc(lat=53.0, lon=-3.0), run["conditions"]) self.assertTrue(aRunAction.score) def testRunActivity(self): aRunAction = actions.GaussDistFromIdeal(datetime.datetime.now(tz=pytz.UTC), Loc(lat=53.0, lon=-3.0), run["conditions"]) possibility = [aRunAction] * 5 aRunActivity = Activity(possibility) self.assertTrue(aRunActivity.score)
import cPickle as pickle import datetime import os import pytz import unittest import isodate import dre.actions as actions from dre.when_decision import * from dre.what_decision import * from dre.decision import * from dre.forecast_cache import ForecastCache from database import database run = database.get_default_values_conf("tests")['default_values']["run"] run.update(database.get_default_values_conf("tests")['general_config']["run"]) sunbathe = database.get_default_values_conf("tests")['default_values']["sunbathe"] sunbathe.update(database.get_default_values_conf("tests")['general_config']["sunbathe"]) cinema = database.get_default_values_conf("tests")['default_values']["cinema"] cinema.update(database.get_default_values_conf("tests")['general_config']["cinema"]) class WhenDecisionTest(unittest.TestCase): cache = ForecastCache() base = os.path.split(__file__)[0] with open(os.path.join(base, 'data', 'testForecast.pkl'), "rb") as f: timesteps = pickle.load(f) cache.cache_forecast(timesteps, Loc(lat=53.0, lon=-3.0))
import unittest import datetime import pytz import sys sys.path.append("..") from dre.decision import * import dre.actions as actions from database import database run = database.get_default_values_conf("tests")['general_config']["run"] class RunTest(unittest.TestCase): def testRunAction(self): aRunAction = actions.GaussDistFromIdeal( datetime.datetime.now(tz=pytz.UTC), Loc(lat=53.0, lon=-3.0), run["conditions"]) self.assertTrue(aRunAction.score) def testRunActivity(self): aRunAction = actions.GaussDistFromIdeal( datetime.datetime.now(tz=pytz.UTC), Loc(lat=53.0, lon=-3.0), run["conditions"]) possibility = [aRunAction] * 5 aRunActivity = Activity(possibility) self.assertTrue(aRunActivity.score)