FilesExpand file tree

master

/

test_gitutils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

85 lines (67 loc) · 2.65 KB

master

/

test_gitutils.py

Copy path

Top

File metadata and controls

• Code
• Blame

85 lines (67 loc) · 2.65 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

#!/usr/bin/env python3

from pathlib import Path

from unittest import main, SkipTest, TestCase

from gitutils import (

_shasum,

are_ghstack_branches_in_sync,

GitRepo,

patterns_to_regex,

PeekableIterator,

)

BASE_DIR = Path(__file__).parent

class TestPeekableIterator(TestCase):

def test_iterator(self, input_: str = "abcdef") -> None:

iter_ = PeekableIterator(input_)

for idx, c in enumerate(iter_):

self.assertEqual(c, input_[idx])

def test_is_iterable(self) -> None:

from collections.abc import Iterator

iter_ = PeekableIterator("")

self.assertTrue(isinstance(iter_, Iterator))

def test_peek(self, input_: str = "abcdef") -> None:

iter_ = PeekableIterator(input_)

for idx, c in enumerate(iter_):

if idx + 1 < len(input_):

self.assertEqual(iter_.peek(), input_[idx + 1])

else:

self.assertTrue(iter_.peek() is None)

class TestPattern(TestCase):

def test_double_asterisks(self) -> None:

allowed_patterns = [

"aten/src/ATen/native/**LinearAlgebra*",

]

patterns_re = patterns_to_regex(allowed_patterns)

fnames = [

"aten/src/ATen/native/LinearAlgebra.cpp",

"aten/src/ATen/native/cpu/LinearAlgebraKernel.cpp",

]

for filename in fnames:

self.assertTrue(patterns_re.match(filename))

class TestGitRepo(TestCase):

def setUp(self) -> None:

repo_dir = BASE_DIR.parent.parent.absolute()

if not (repo_dir / ".git").is_dir():

raise SkipTest(

"Can't find git directory, make sure to run this test on real repo checkout"

)

self.repo = GitRepo(str(repo_dir))

def _skip_if_ref_does_not_exist(self, ref: str) -> None:

"""Skip test if ref is missing as stale branches are deleted with time"""

try:

self.repo.show_ref(ref)

except RuntimeError as e:

raise SkipTest(f"Can't find head ref {ref} due to {str(e)}") from e

def test_compute_diff(self) -> None:

diff = self.repo.diff("HEAD")

sha = _shasum(diff)

self.assertEqual(len(sha), 64)

def test_ghstack_branches_in_sync(self) -> None:

head_ref = "gh/SS-JIA/206/head"

self._skip_if_ref_does_not_exist(head_ref)

self.assertTrue(are_ghstack_branches_in_sync(self.repo, head_ref))

def test_ghstack_branches_not_in_sync(self) -> None:

head_ref = "gh/clee2000/1/head"

self._skip_if_ref_does_not_exist(head_ref)

self.assertFalse(are_ghstack_branches_in_sync(self.repo, head_ref))

if __name__ == "__main__":

main()