known first parties

pyproject.toml

@@ -5,6 +5,7 @@ exclude = "gui/mozregui/ui"
 [tool.isort]
 line_length = 100
 skip_glob = "**/gui/mozregui/ui/*"
+known_first_party=mozregression,mozregui
 # For compatibility with black:
 multi_line_output = 3
 include_trailing_comma = "True"

tests/unit/conftest.py

@@ -0,0 +1,20 @@
+import pytest
+
+from mozregression import build_range
+from mozregression.fetch_build_info import InfoFetcher
+
+
+class RangeCreator(object):
+    def __init__(self, mocker):
+        self.mocker = mocker
+
+    def create(self, values):
+        info_fetcher = self.mocker.Mock(spec=InfoFetcher)
+        info_fetcher.find_build_info.side_effect = lambda i: i
+        future_build_infos = [build_range.FutureBuildInfo(info_fetcher, v) for v in values]
+        return build_range.BuildRange(info_fetcher, future_build_infos)
+
+
+@pytest.fixture
+def range_creator(mocker):
+    return RangeCreator(mocker)