示例#1
0
    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)