test/models_selection_test.py

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