From 24ff6a703e63af5d71f7f93e3e7f2086cd847b6d Mon Sep 17 00:00:00 2001
From: Tanner Prestegard <tanner.prestegard@ligo.org>
Date: Thu, 6 Jun 2019 08:55:21 -0500
Subject: [PATCH] Add a few more "natural languange" unit tests for search
queries
---
gracedb/search/tests/test_queries.py | 15 +++++++++++----
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/gracedb/search/tests/test_queries.py b/gracedb/search/tests/test_queries.py
index a9fbd810b..b255275b3 100644
--- a/gracedb/search/tests/test_queries.py
+++ b/gracedb/search/tests/test_queries.py
@@ -111,10 +111,17 @@ SUPEREVENT_QUERY_TEST_DATA = [
SERVER_TZ.localize(datetime.datetime(2019, 5, 5, 12, 34, 56))]) &
DEFAULT_Q),
('yesterday .. now', Q(created__range=[
- SERVER_TZ.localize(datetime.datetime(2019, 3, 30, 0, 0, 0)),
- MOCK_NOW_DT]) & DEFAULT_Q),
- ('noon', Q(created=SERVER_TZ.localize(datetime.datetime(2019, 3, 31,
- 12, 0, 0))) & DEFAULT_Q),
+ MOCK_NOW_DT.replace(day=MOCK_NOW_DT.day-1, hour=0, minute=0,
+ second=0, microsecond=0), MOCK_NOW_DT]) & DEFAULT_Q),
+ ('a couple of days ago', Q(created=MOCK_NOW_DT.replace(
+ day=MOCK_NOW_DT.day-2)) & DEFAULT_Q),
+ ('1 week ago .. now', Q(created__range=[
+ MOCK_NOW_DT - datetime.timedelta(days=7), MOCK_NOW_DT]) & DEFAULT_Q),
+ ('3 days ago .. 2 days ago', Q(created__range=[
+ MOCK_NOW_DT - datetime.timedelta(days=3),
+ MOCK_NOW_DT - datetime.timedelta(days=2)]) & DEFAULT_Q),
+ ('noon', Q(created=MOCK_NOW_DT.replace(hour=12, minute=0, second=0,
+ microsecond=0)) & DEFAULT_Q),
('far <= 1e-10', Q(preferred_event__far__lte=1e-10) & DEFAULT_Q),
('far < 1e-10', Q(preferred_event__far__lt=1e-10) & DEFAULT_Q),
('far > 1e-10', Q(preferred_event__far__gt=1e-10) & DEFAULT_Q),
--
GitLab