class TestDataFileListItem(TestFileListItem): """ Subclassing TestFileListItem will also run those tests too. """ def setUp(self): self.params = {'hi': 9, 'bye': 19} TestFileListItem.setUp(self) def _set_item(self): self.item = DataFileListItem(self.filename, self.params) def test_data_file_list_item_params(self): params = self.item.get_params() self.assertEqual(params, self.params) # Test immutable params['qwertyuiop'] = 1 self.assertNotIn('qwertyuiop', self.item.get_params()) def test_data_file_list_item_get_param(self): self.assertEqual(self.item.get_param('hi'), 9) self.assertEqual(self.item.get_param('bye'), 19) self.assertIsNone(self.item.get_param('sdafkfasfweoaifeawfksfd'))
def _set_item(self): self.item = DataFileListItem(self.filename, self.params)