diff trafficintelligence/tests/events.txt @ 1148:eb88d2984637

corrected interaction classification
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 12 May 2020 01:16:54 -0400
parents aafbc0bab925
children 14140b55e580
line wrap: on
line diff
--- a/trafficintelligence/tests/events.txt	Wed May 06 00:58:21 2020 -0400
+++ b/trafficintelligence/tests/events.txt	Tue May 12 01:16:54 2020 -0400
@@ -32,3 +32,41 @@
 True
 >>> inter.getIndicator(Interaction.indicatorNames[1])[6] # doctest:+ELLIPSIS
 3.1415...
+
+>>> from collections import Counter
+>>> from numpy import pi
+>>> o1 = MovingObject.generate(0, Point(0,0), Point(1,0), TimeInterval(0,100))
+>>> o2 = MovingObject.generate(0, Point(100,1), Point(-1,0), TimeInterval(0,100))
+>>> inter12 = Interaction(roadUser1 = o1, roadUser2 = o2)
+>>> inter12.computeIndicators()
+>>> inter12.categorize(pi*20/180, pi*60/180)
+>>> Counter(inter12.categories.values()).most_common()[0][0] # head on
+0
+>>> inter12.categories[max(inter12.categories.keys())] # then side
+2
+>>> o3 = MovingObject.generate(0, Point(0,2), Point(1,0), TimeInterval(0,100))
+>>> inter13 = Interaction(roadUser1 = o1, roadUser2 = o3)
+>>> inter13.computeIndicators()
+>>> inter13.categorize(pi*20/180, pi*60/180)
+>>> Counter(inter13.categories.values()).most_common()[0][0] # parallel
+3
+>>> len(Counter(inter13.categories.values()))
+1
+>>> o4 = MovingObject.generate(0, Point(100,20), Point(-1,0), TimeInterval(0,100))
+>>> inter14 = Interaction(roadUser1 = o1, roadUser2 = o4)
+>>> inter14.computeIndicators()
+>>> inter14.categorize(pi*20/180, pi*60/180)
+>>> Counter(inter14.categories.values()).most_common()[0][0] # side
+2
+>>> inter12.categories[0] # first head one
+0
+>>> inter12.categories[max(inter12.categories.keys())] # then side
+2
+>>> o5 = MovingObject.generate(0, Point(50,50), Point(0,-1), TimeInterval(0,100))
+>>> inter15 = Interaction(roadUser1 = o1, roadUser2 = o5)
+>>> inter15.computeIndicators()
+>>> inter15.categorize(pi*20/180, pi*60/180)
+>>> Counter(inter15.categories.values()).most_common()[0][0] # side
+2
+>>> len(Counter(inter15.categories.values()))
+1