def test__get_data(self, build_granule_mock, TaxyTool_mock, RecordDictionaryTool_mock): config = { 'constraints': { 'count': 6 }, 'max_records': 4, 'data_producer_id': sentinel.dprod_id, 'taxonomy': sentinel.taxy_tool } TaxyTool_mock.load.return_value = sentinel.ttool_ret_val RecordDictionaryTool_mock.side_effect = lambda **kwargs: {} for x in FibonacciDataHandler._get_data(config): RecordDictionaryTool_mock.assert_called_with( taxonomy=sentinel.ttool_ret_val) build_granule_mock.assert_called_with( taxonomy=sentinel.ttool_ret_val, data_producer_id=sentinel.dprod_id, record_dictionary=ANY) cargs = build_granule_mock.call_args self.assertIn('data', cargs[1]['record_dictionary']) self.assertIsInstance(cargs[1]['record_dictionary']['data'], numpy.ndarray) self.assertEquals(build_granule_mock.call_count, 2) self.assertEquals(RecordDictionaryTool_mock.call_count, 2) TaxyTool_mock.load.assert_called_once_with(sentinel.taxy_tool)
def test__get_data(self, build_granule_mock, ParamDict_mock, RecordDictionaryTool_mock): config={'constraints':{'count':6},'max_records':4,'data_producer_id':sentinel.dprod_id,'param_dictionary':sentinel.pdict} ParamDict_mock.load.return_value = sentinel.pdict_return_value RecordDictionaryTool_mock.side_effect = lambda **kwargs: {} for x in FibonacciDataHandler._get_data(config): RecordDictionaryTool_mock.assert_called_with(param_dictionary=sentinel.pdict_return_value) build_granule_mock.assert_called_with(data_producer_id=sentinel.dprod_id, record_dictionary=ANY, param_dictionary=sentinel.pdict_return_value) cargs=build_granule_mock.call_args self.assertIn('data', cargs[1]['record_dictionary']) self.assertIsInstance(cargs[1]['record_dictionary']['data'], numpy.ndarray) self.assertEquals(build_granule_mock.call_count, 2) self.assertEquals(RecordDictionaryTool_mock.call_count, 2) ParamDict_mock.load.assert_called_once_with(sentinel.pdict)
def test__get_data(self, RecordDictionaryTool_mock): config = {'constraints': {'count': 6}, 'max_records': 4, 'data_producer_id': sentinel.dprod_id, 'param_dictionary': sentinel.pdict, 'stream_def': sentinel.stream_def_id } my_dict = {} def setitem(name, val): my_dict[name] = val retval = MagicMock(spec=RecordDictionaryTool) retval.__setitem__ = Mock(side_effect=setitem) retval.to_granule.return_value = MagicMock(spec=Granule) RecordDictionaryTool_mock.return_value = retval for x in FibonacciDataHandler._get_data(config): self.assertTrue(isinstance(x, Granule)) retval.to_granule.assert_any_call()
def test__get_data(self, build_granule_mock, ParamDict_mock, RecordDictionaryTool_mock): config = { "constraints": {"count": 6}, "max_records": 4, "data_producer_id": sentinel.dprod_id, "param_dictionary": sentinel.pdict, } ParamDict_mock.load.return_value = sentinel.pdict_return_value RecordDictionaryTool_mock.side_effect = lambda **kwargs: {} for x in FibonacciDataHandler._get_data(config): RecordDictionaryTool_mock.assert_called_with(param_dictionary=sentinel.pdict_return_value) build_granule_mock.assert_called_with( data_producer_id=sentinel.dprod_id, record_dictionary=ANY, param_dictionary=sentinel.pdict_return_value ) cargs = build_granule_mock.call_args self.assertIn("data", cargs[1]["record_dictionary"]) self.assertIsInstance(cargs[1]["record_dictionary"]["data"], numpy.ndarray) self.assertEquals(build_granule_mock.call_count, 2) self.assertEquals(RecordDictionaryTool_mock.call_count, 2) ParamDict_mock.load.assert_called_once_with(sentinel.pdict)
def test__constraints_for_new_request(self): ret = FibonacciDataHandler._constraints_for_new_request({}) self.assertTrue("count" in ret) i = ret["count"] self.assertIsInstance(i, int) self.assertTrue(5 <= i <= 20)
def test__init_acquisition_cycle(self, _init_acquisition_cycle_mock): config = {} FibonacciDataHandler._init_acquisition_cycle(config) _init_acquisition_cycle_mock.assert_called_once_with(config)
def test__constraints_for_new_request(self): ret = FibonacciDataHandler._constraints_for_new_request({}) self.assertTrue('count' in ret) i = ret['count'] self.assertIsInstance(i, int) self.assertTrue(5 <= i <= 20)
def test__new_data_constraints(self): ret = FibonacciDataHandler._new_data_constraints({}) self.assertTrue('count' in ret) i=ret['count'] self.assertIsInstance(i, int) self.assertTrue(5 <= i <= 20)