mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
8549fafd36
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78189 Approved by: https://github.com/soulitzer
46 lines
2.0 KiB
Python
46 lines
2.0 KiB
Python
import unittest
|
|
import tempfile
|
|
from commitlist import CommitList
|
|
|
|
class TestCommitList(unittest.TestCase):
|
|
def test_create_new(self):
|
|
with tempfile.TemporaryDirectory() as tempdir:
|
|
commit_list_path = f'{tempdir}/commitlist.csv'
|
|
commit_list = CommitList.create_new(commit_list_path, 'v1.5.0', '6000dca5df')
|
|
self.assertEqual(len(commit_list.commits), 33)
|
|
self.assertEqual(commit_list.commits[0].commit_hash, '7335f079ab')
|
|
self.assertTrue(commit_list.commits[0].title.startswith('[pt][quant] qmul and qadd'))
|
|
self.assertEqual(commit_list.commits[-1].commit_hash, '6000dca5df')
|
|
self.assertTrue(commit_list.commits[-1].title.startswith('[nomnigraph] Copy device option when customize '))
|
|
|
|
def test_read_write(self):
|
|
with tempfile.TemporaryDirectory() as tempdir:
|
|
commit_list_path = f'{tempdir}/commitlist.csv'
|
|
initial = CommitList.create_new(commit_list_path, 'v1.5.0', '7543e7e558')
|
|
initial.write_to_disk()
|
|
|
|
expected = CommitList.from_existing(commit_list_path)
|
|
expected.commits[-2].category = 'foobar'
|
|
expected.write_to_disk()
|
|
|
|
commit_list = CommitList.from_existing(commit_list_path)
|
|
for commit, expected in zip(commit_list.commits, expected.commits):
|
|
self.assertEqual(commit, expected)
|
|
|
|
def test_update_to(self):
|
|
with tempfile.TemporaryDirectory() as tempdir:
|
|
commit_list_path = f'{tempdir}/commitlist.csv'
|
|
initial = CommitList.create_new(commit_list_path, 'v1.5.0', '7543e7e558')
|
|
initial.commits[-2].category = 'foobar'
|
|
self.assertEqual(len(initial.commits), 2143)
|
|
initial.write_to_disk()
|
|
|
|
commit_list = CommitList.from_existing(commit_list_path)
|
|
commit_list.update_to('5702a28b26')
|
|
self.assertEqual(len(commit_list.commits), 2143 + 4)
|
|
self.assertEqual(commit_list.commits[-5], initial.commits[-1])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|