Skip to content

Commit

Permalink
Fix incorrect query-result type imports.
Browse files Browse the repository at this point in the history
This test wanted the result objects from

   lsst.daf.butler.registry.queries

for use in annotations, but it was getting them from

   lsst.daf.butler.queries

and we didn't notice because we don't run MyPy on tests.
  • Loading branch information
TallJimbo committed Mar 28, 2024
1 parent ec40983 commit c3b43a4
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/test_query_relations.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,12 @@
import re
import unittest

from lsst.daf.butler import DataCoordinateQueryResults, DatasetRefQueryResults, DimensionRecordQueryResults
from lsst.daf.butler.registry import MissingSpatialOverlapError, RegistryConfig, _RegistryFactory
from lsst.daf.butler.registry.queries import (
DataCoordinateQueryResults,
DatasetQueryResults,
DimensionRecordQueryResults,
)
from lsst.daf.butler.transfers import YamlRepoImportBackend

TESTDIR = os.path.abspath(os.path.dirname(__file__))
Expand Down Expand Up @@ -99,7 +103,7 @@ def setUpClass(cls) -> None:
def assert_relation_str(
self,
expected: str,
*results: DataCoordinateQueryResults | DimensionRecordQueryResults | DatasetRefQueryResults,
*results: DataCoordinateQueryResults | DimensionRecordQueryResults | DatasetQueryResults,
) -> None:
"""Assert that checks that one or more registry
queries have relation trees that match the given string.
Expand Down

0 comments on commit c3b43a4

Please sign in to comment.