def test_get_data(self): SUT = CollectData(self.wind_fields) SUT.add_data(self.data) collected_data = SUT.get_data() self.assertDictEqual(collected_data, self.data)
def test_get_data(self): with mock.patch('user.MQTTSubscribe.weewx.units.to_std_system' ) as mock_to_std_system: SUT = CollectData(self.wind_fields, self.units) mock_to_std_system.return_value = dict(self.data) SUT.add_data(self.wind_field, self.data) collected_data = SUT.get_data() self.assertDictEqual(collected_data, self.data)