def test_get_locations(self): self.assertEqual({'foo': ''}, _get_locations(['foo'], {})) self.assertEqual({'foo': 'src'}, _get_locations(['foo'], {'': 'src'})) self.assertEqual({'foo': 'src', 'foo.bar': 'src'}, _get_locations(['foo.bar'], {'': 'src'})) self.assertEqual({'foo': 'src'}, _get_locations(['foo'], {'foo': 'src'})) self.assertEqual({'foo': 'src', 'foo.bar': 'src'}, _get_locations(['foo.bar'], {'foo': 'src'}))