view tests/test-rust-revlog.py @ 43966:f91834ecfdfd

narrow: move `testedwith` after module imports

PyCharm flagged this as a PEP 8 violation, for module level imports not being at
the top of the file.

Differential Revision: https://phab.mercurial-scm.org/D7734

from __future__ import absolute_import
import unittest

try:
    from mercurial import rustext

    rustext.__name__  # trigger immediate actual import
except ImportError:
    rustext = None
else:
    from mercurial.rustext import revlog

from mercurial.testing import revlog as revlogtesting


@unittest.skipIf(
    rustext is None, "rustext module revlog relies on is not available",
)
class RustRevlogIndexTest(revlogtesting.RevlogBasedTestBase):
    def test_heads(self):
        idx = self.parseindex()
        rustidx = revlog.MixedIndex(idx)
        self.assertEqual(rustidx.headrevs(), idx.headrevs())

    def test_len(self):
        idx = self.parseindex()
        rustidx = revlog.MixedIndex(idx)
        self.assertEqual(len(rustidx), len(idx))


if __name__ == '__main__':
    import silenttestrunner

    silenttestrunner.main(__name__)