def test_coalesce_new_atoms_with_no_atom_times(self):
new_atoms = self._mock_atoms(['atom_1', 'atom_2', 'atom_3'])
old_atoms_with_times = {}
project_directory = 'some_project_directory'
atom_grouper = TimeBasedAtomGrouper(new_atoms, 3, old_atoms_with_times, project_directory)
with self.assertRaises(_AtomTimingDataError):
atom_grouper._set_expected_atom_times(new_atoms, old_atoms_with_times, project_directory)
def test_coalesce_new_atoms_with_no_atom_times(self):
new_atoms = self._mock_atoms(['atom_1', 'atom_2', 'atom_3'])
old_atoms_with_times = {}
project_directory = 'some_project_directory'
atom_grouper = TimeBasedAtomGrouper(new_atoms, 3, old_atoms_with_times,
project_directory)
with self.assertRaises(_AtomTimingDataError):
atom_grouper._set_expected_atom_times(new_atoms,
old_atoms_with_times,
project_directory)
def test_coalesce_new_atoms_with_some_atom_times(self):
new_atoms = self._mock_atoms(['atom_2', 'atom_3', 'atom_4', 'atom_5'])
old_atoms_with_times = {'atom_1': 1.0, 'atom_2': 2.0, 'atom_3': 3.0}
expected_contents = {'atom_2': 2.0, 'atom_3': 3.0, 'atom_4': 3.0, 'atom_5': 3.0}
atom_grouper = TimeBasedAtomGrouper(new_atoms, 3, old_atoms_with_times, 'some_project_directory')
total_time = atom_grouper._set_expected_atom_times(new_atoms, old_atoms_with_times, 'some_project_directory')
self.assertEquals(total_time, 11.0)
self._assert_coalesced_contents(new_atoms, expected_contents)
def test_coalesce_new_atoms_with_all_atom_times(self):
new_atoms = self._mock_atoms(['atom_1', 'atom_2', 'atom_3'])
old_atoms_with_times = {'atom_1': 1.0, 'atom_2': 2.0, 'atom_3': 3.0}
expected_contents = {'atom_1': 1.0, 'atom_2': 2.0, 'atom_3': 3.0}
atom_grouper = TimeBasedAtomGrouper(new_atoms, 3, old_atoms_with_times,
'some_project_directory')
total_time = atom_grouper._set_expected_atom_times(
new_atoms, old_atoms_with_times, 'some_project_directory')
self.assertEquals(total_time, 6.0)
self._assert_coalesced_contents(new_atoms, expected_contents)
