def test_run_off_rep_5(self, mock_off, mock_rep): # run neither indexer; input a dict # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm.settings(1, 'offset interval', 0) workm.settings(1, 'filter repeats', False) in_val = {'b': 43, 'a': 42, 'c': 44} # run actual = workm._run_off_rep(1, in_val) # test self.assertSequenceEqual(in_val, actual) self.assertEqual(0, workm._data[1].get_data.call_count)
def test_run_off_rep_1(self): """run neither indexer""" # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm.settings(1, 'offset interval', 0) workm.settings(1, 'filter repeats', False) in_val = 42 # run actual = workm._run_off_rep(1, in_val) # test self.assertEqual(in_val, actual) self.assertEqual(0, workm._data[1].get_data.call_count)
def test_run_off_rep_3(self, mock_off, mock_rep): # run repeat indexer # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm._data[1].get_data.return_value = 24 workm.settings(1, 'offset interval', 0) workm.settings(1, 'filter repeats', True) in_val = 42 # run actual = workm._run_off_rep(1, in_val) # test self.assertEqual(workm._data[1].get_data.return_value, actual) workm._data[1].get_data.assert_called_once_with([mock_rep], {}, in_val)
def test_run_off_rep_3(self): """run repeat indexer""" # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm._data[1].get_data.return_value = 24 workm.settings(1, 'offset interval', 0) workm.settings(1, 'filter repeats', True) in_val = 42 exp_analyzers = [repeat.FilterByRepeatIndexer] # run actual = workm._run_off_rep(1, in_val) # test self.assertEqual(workm._data[1].get_data.return_value, actual) workm._data[1].get_data.assert_called_once_with(exp_analyzers, {}, in_val)
def test_run_off_rep_6(self, mock_off, mock_rep): # run offset indexer with is_horizontal set to True # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm._data[1].get_data.return_value = 24 workm.settings(1, 'offset interval', 0.5) workm.settings(1, 'filter repeats', False) in_val = 42 # run actual = workm._run_off_rep(1, in_val, True) # test self.assertEqual(workm._data[1].get_data.return_value, actual) workm._data[1].get_data.assert_called_once_with([mock_off], {'quarterLength': 0.5, 'method': None}, in_val)
def test_run_off_rep_4(self, mock_off, mock_rep): # run offset and repeat indexer # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm._data[1].get_data.return_value = 24 workm.settings(1, 'offset interval', 0.5) workm.settings(1, 'filter repeats', True) in_val = 42 # run actual = workm._run_off_rep(1, in_val) # test self.assertEqual(workm._data[1].get_data.return_value, actual) self.assertEqual(2, workm._data[1].get_data.call_count) workm._data[1].get_data.assert_any_call([mock_off], {'quarterLength': 0.5}, in_val) workm._data[1].get_data.assert_any_call([mock_rep], {}, workm._data[1].get_data.return_value)
def test_run_off_rep_4(self): """run offset and repeat indexer""" # setup workm = WorkflowManager(['', '', '']) workm._data = [None, MagicMock(spec=IndexedPiece), None] workm._data[1].get_data.return_value = 24 workm.settings(1, 'offset interval', 0.5) workm.settings(1, 'filter repeats', True) in_val = 42 exp_an_off = [offset.FilterByOffsetIndexer] exp_an_rep = [repeat.FilterByRepeatIndexer] # run actual = workm._run_off_rep(1, in_val) # test self.assertEqual(workm._data[1].get_data.return_value, actual) self.assertEqual(2, workm._data[1].get_data.call_count) workm._data[1].get_data.assert_any_call(exp_an_off, {'quarterLength': 0.5}, in_val) workm._data[1].get_data.assert_any_call(exp_an_rep, {}, workm._data[1].get_data.return_value)