suppress terminal and reverse predictions for RL (#683)

mbta · Aug 31, 2023 · 097ea6c · 097ea6c
1 parent 7f1dc9f
commit 097ea6c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 76 deletions.
diff --git a/lib/predictions/predictions.ex b/lib/predictions/predictions.ex
@@ -151,16 +151,11 @@ defmodule Predictions.Predictions do
   end
 
   @spec sufficient_certainty?(map(), String.t()) :: boolean()
-  defp sufficient_certainty?(_stop_time_event, route_id)
-       when route_id in ["Mattapan", "Green-B", "Green-C", "Green-D", "Green-E"] do
-    true
+  defp sufficient_certainty?(stop_time_event, route_id) when route_id in ["Red"] do
+    is_nil(stop_time_event["uncertainty"]) or stop_time_event["uncertainty"] < 120
   end
 
-  defp sufficient_certainty?(stop_time_event, _route_id) do
-    if Application.get_env(:realtime_signs, :filter_uncertain_predictions?) do
-      is_nil(stop_time_event["uncertainty"]) or stop_time_event["uncertainty"] <= 300
-    else
-      true
-    end
+  defp sufficient_certainty?(_stop_time_event, _route_id) do
+    true
   end
 end
diff --git a/test/predictions/predictions_test.exs b/test/predictions/predictions_test.exs
@@ -882,7 +882,7 @@ defmodule Predictions.PredictionsTest do
               "timestamp" => nil,
               "trip" => %{
                 "direction_id" => 0,
-                "route_id" => "Blue",
+                "route_id" => "Red",
                 "schedule_relationship" => "SCHEDULED",
                 "start_date" => "20170329",
                 "start_time" => nil,
@@ -909,72 +909,6 @@ defmodule Predictions.PredictionsTest do
       assert predictions_map == %{}
     end
 
-    test "doesn't filter predictions with high uncertainty when feature is off" do
-      reassign_env(:filter_uncertain_predictions?, false)
-
-      feed_message = %{
-        "entity" => [
-          %{
-            "alert" => nil,
-            "id" => "1490783458_32568935",
-            "is_deleted" => false,
-            "trip_update" => %{
-              "delay" => nil,
-              "stop_time_update" => [
-                %{
-                  "arrival" => %{
-                    "delay" => nil,
-                    "time" => 1_491_570_110,
-                    "uncertainty" => 360
-                  },
-                  "departure" => %{
-                    "delay" => nil,
-                    "time" => 1_491_570_120,
-                    "uncertainty" => 360
-                  },
-                  "schedule_relationship" => "SCHEDULED",
-                  "stop_id" => "70063",
-                  "stop_sequence" => 1,
-                  "stops_away" => 1,
-                  "stopped?" => true,
-                  "boarding_status" => "Stopped 1 stop away"
-                }
-              ],
-              "timestamp" => nil,
-              "trip" => %{
-                "direction_id" => 0,
-                "route_id" => "Red",
-                "schedule_relationship" => "SCHEDULED",
-                "start_date" => "20170329",
-                "start_time" => nil,
-                "trip_id" => "32568935"
-              },
-              "vehicle" => %{
-                "id" => "R-54639F6C",
-                "label" => "1631",
-                "license_plate" => nil
-              }
-            },
-            "vehicle" => nil
-          }
-        ],
-        "header" => %{
-          "gtfs_realtime_version" => "1.0",
-          "incrementality" => "FULL_DATASET",
-          "timestamp" => 1_490_783_458
-        }
-      }
-
-      assert {%{
-                {"70063", 0} => [
-                  %Predictions.Prediction{
-                    seconds_until_arrival: 110,
-                    seconds_until_departure: 120
-                  }
-                ]
-              }, _} = get_all(feed_message, @current_time)
-    end
-
     test "doesn't filter out uncertain light rail predictions" do
       reassign_env(:filter_uncertain_predictions?, true)