test/models_selection_test.py

   1 import unittest
   2
   3 import cola.models.selection as selection
   4
   5
   6 class T(object):
   7     pass
   8
   9
  10 class SelectionTestCase(unittest.TestCase):
  11
  12     def test_union(self):
  13         t = T()
  14         t.staged = ['a']
  15         t.unmerged = ['a', 'b']
  16         t.modified = ['b', 'a', 'c']
  17         t.untracked = ['d']
  18
  19         expect = ['a', 'b', 'c', 'd']
  20         actual = selection.union(t)
  21         self.assertEqual(expect, actual)
  22
  23
  24 if __name__ == '__main__':
  25     unittest.main()