Exemple #1
0
    def test_end(self):
        i = Interval(Time(2013, 1, 1), Time(2013, 1, 2))
        self.assertEqual(i.end, Time(2013, 1, 2, 23, 59))

        i = Interval(None, Time(2013, 1, 2))
        self.assertEqual(i.end, Time(2013, 1, 2, 23, 59))

        i = Interval(Time(2013, 1, 1), None)
        self.assertIsNone(i.end)
Exemple #2
0
    def test_start(self):
        i = Interval(Time(2013, 1, 1), Time(2013, 1, 2))
        self.assertEqual(i.start, Time(2013, 1, 1, 0, 0))

        i = Interval(Time(2013, 1, 1), None)
        self.assertEqual(i.start, Time(2013, 1, 1, 0, 0))

        i = Interval(None, Time(2013, 1, 2))
        self.assertIsNone(i.start)
Exemple #3
0
def test_latent_time_interval():
    parse = ctparse("8:00 pm - 9:00 pm",
                    ts=datetime(2020, 1, 1, 7, 0),
                    latent_time=False)
    assert parse
    assert parse.resolution == Interval(Time(None, None, None, 20, 00),
                                        Time(None, None, None, 21, 00))

    parse = ctparse("8:00 pm - 9:00 pm",
                    ts=datetime(2020, 1, 1, 7, 0),
                    latent_time=True)
    assert parse
    assert parse.resolution == Interval(Time(2020, 1, 1, 20, 00),
                                        Time(2020, 1, 1, 21, 00))
Exemple #4
0
def test_dummy():
    scorer = DummyScorer()
    pp = PartialParse((Time(), Interval()), ("rule1", "rule2"))

    assert scorer.score("a", datetime.datetime(2019, 1, 1), pp) == 0.0
    assert scorer.score_final("a", datetime.datetime(2019, 1, 1), pp,
                              pp.prod[0]) == 0.0
def _latent_time_interval(ts: datetime, ti: Interval) -> Interval:
    assert ti.t_from and ti.t_to  # guaranteed by the caller
    dm_from = ts + relativedelta(hour=ti.t_from.hour,
                                 minute=ti.t_from.minute or 0)
    dm_to = ts + relativedelta(hour=ti.t_to.hour, minute=ti.t_to.minute or 0)
    if dm_from <= ts:
        dm_from += relativedelta(days=1)
        dm_to += relativedelta(days=1)
    return Interval(
        t_from=Time(
            year=dm_from.year,
            month=dm_from.month,
            day=dm_from.day,
            hour=dm_from.hour,
            minute=dm_from.minute,
            period=ti.t_from.period,
        ),
        t_to=Time(
            year=dm_to.year,
            month=dm_to.month,
            day=dm_to.day,
            hour=dm_to.hour,
            minute=dm_to.minute,
            period=ti.t_to.period,
        ),
    )
Exemple #6
0
def test_random():
    rng = random.Random(42)
    scorer = RandomScorer(rng)

    pp = PartialParse((Time(), Interval()), ("rule1", "rule2"))

    assert 0.0 <= scorer.score("a", datetime.datetime(2019, 1, 1), pp) <= 1.0
    assert (0.0 <= scorer.score_final("a", datetime.datetime(2019, 1, 1), pp,
                                      pp.prod[1]) <= 1.0)
Exemple #7
0
    def test_eq(self):
        a = Artifact()
        b = Artifact()
        self.assertEqual(a, b)

        a = Time(2017, 12, 12, 12, 12, 4, 'morning')
        b = Time(2017, 12, 12, 12, 12, 4, 'morning')
        self.assertEqual(a, b)

        a = Time(2017, 12, 12, 12, 12, 4, 'morning')
        b = Time(2017, 12, 12, 12, 12, 3, 'morning')
        self.assertNotEqual(a, b)

        a = Time()
        b = Interval()
        self.assertNotEqual(a, b)
Exemple #8
0
    def test_from_str(self):
        # Complete interval
        t1 = Time(year=1, month=1, day=1, hour=1, minute=1, DOW=1, POD='pod')
        t2 = Time(year=2, month=1, day=1, hour=1, minute=1, DOW=1, POD='pod')
        interval = Interval(t1, t2)
        i_back = Interval.from_str(str(interval))
        self.assertEqual(interval, i_back)

        # Incomplete interval
        interval = Interval(None, t2)
        i_back = Interval.from_str(str(interval))
        self.assertEqual(interval, i_back)

        # Zeroed interval
        interval = Interval()
        i_back = Interval.from_str(str(interval))
        self.assertEqual(interval, i_back)

        # Mistake
        with self.assertRaises(ValueError):
            Interval.from_str('X-X-X X: X(X/X) -X-X-X X: X(X/X)')
Exemple #9
0
def test_nbscorer():
    # We only test that it runs just fine
    X = [("a", "b"), ("a", ), ("b"), ("a", "b", "a", "b")]
    y = [False, True, True, False]

    model = train_naive_bayes(X, y)
    scorer = NaiveBayesScorer(model)

    pp = PartialParse((Time(), Interval()), ("rule1", "rule2"))

    pp.prod[0].mstart = 0
    pp.prod[1].mend = 1

    pp.prod[0].mend = 1
    pp.prod[1].mend = 2

    assert 0.0 <= scorer.score("ab", datetime.datetime(2019, 1, 1), pp) <= 1.0
    assert (0.0 <= scorer.score_final("ab", datetime.datetime(2019, 1, 1), pp,
                                      pp.prod[1]) <= 1.0)
Exemple #10
0
def test_parse_nb_string() -> None:
    t = Time(year=1, month=1, day=1, hour=1, minute=1, DOW=1, POD="pod")

    assert t == parse_nb_string("Time[]{0001-01-01 01:01 (1/pod)}")
    assert Interval(Time(), Time()) == parse_nb_string(
        "Interval[]{X-X-X X:X (X/X) - X-X-X X:X (X/X)}")
Exemple #11
0
 def test_repr(self):
     self.assertEqual(repr(Interval(Time(), Time())),
                      'Interval[0-0]{X-X-X X:X (X/X) - X-X-X X:X (X/X)}')
Exemple #12
0
 def test_isTimeInterval(self):
     self.assertTrue(Interval(Time(hour=1), Time(hour=2)).isTimeInterval)
Exemple #13
0
 def test_init(self):
     self.assertIsNotNone(Interval())