def test_get_by_status(self): pds = ProcessedData.get_by_status('sandbox') self.assertEqual(pds, set()) pds = ProcessedData.get_by_status('private') self.assertEqual(pds, set([1])) ProcessedData.create(self.params_table, self.params_id, self.filepaths, preprocessed_data=self.preprocessed_data) pds = ProcessedData.get_by_status('sandbox') self.assertEqual(pds, set([2])) pds = ProcessedData.get_by_status('private') self.assertEqual(pds, set([1]))