python中unique函数_Python pandas 模块，unique() 实例源码 - 编程字典

最新推荐文章于 2024-03-14 22:17:57 发布

weixin_39776298

最新推荐文章于 2024-03-14 22:17:57 发布

阅读量271

点赞数

文章标签： python中unique函数

本文链接：https://blog.csdn.net/weixin_39776298/article/details/113501301

版权

这个博客主要测试了获取地震事件数据的函数，在遇到URL读取错误和数据格式错误时的处理。首先，它验证了在正常情况下，能正确保存前两个事件，并且数据长度为2。然后模拟了HTTP 413错误和格式错误的场景，再次确认结果与之前一致，并检查了错误日志记录。

摘要由CSDN通过智能技术生成

def test_get_events(self, mock_query):

urlread_sideeffect = ["""1|2|3|4|5|6|7|8|9|10|11|12|13

20160508_0000129|2016-05-08 05:17:11.500000|40.57|52.23|60.0|AZER|EMSC-RTS|AZER|505483|ml|3.1|AZER|CASPIAN SEA, OFFSHR TURKMENISTAN

20160508_0000004|2016-05-08 01:45:30.300000|44.96|15.35|2.0|EMSC|EMSC-RTS|EMSC|505183|ml|3.6|EMSC|CROATIA

20160508_0000113|2016-05-08 22:37:20.100000|45.68|26.64|163.0|BUC|EMSC-RTS|BUC|505351|ml|3.4|BUC|ROMANIA

--- ERRROR --- THIS IS MALFORMED 20160508_abc0113|2016-05-08 22:37:20.100000| --- ERROR --- |26.64|163.0|BUC|EMSC-RTS|BUC|505351|ml|3.4|BUC|ROMANIA

"""]

data = self.get_events_df(urlread_sideeffect, self.session, "http://eventws", db_bufsize=self.db_buf_size)

# assert only first two events events were successfully saved

assert len(self.session.query(Event).all()) == len(pd.unique(data['id'])) == 2

# AND data to save has length 2:

assert len(data) == 2

# now download again, with an url error:

urlread_sideeffect = [413, """1|2|3|4|5|6|7|8|9|10|11|12|13

20160508_0000129|2016-05-08 05:17:11.500000|40.57|52.23|60.0|AZER|EMSC-RTS|AZER|505483|ml|3.1|AZER|CASPIAN SEA, OFFSHR TURKMENISTAN

20160508_0000004|2016-05-08 01:45:30.300000|44.96|15.35|2.0|EMSC|EMSC-RTS|EMSC|505183|ml|3.6|EMSC|CROATIA

20160508_0000113|2016-05-08 22:37:20.100000|45.68|26.64|163.0|BUC|EMSC-RTS|BUC|505351|ml|3.4|BUC|ROMANIA

--- ERRROR --- THIS IS MALFORMED 20160508_abc0113|2016-05-08 22:37:20.100000| --- ERROR --- |26.64|163.0|BUC|EMSC-RTS|BUC|505351|ml|3.4|BUC|ROMANIA

""", URLError('blabla23___')]

data = self.get_events_df(urlread_sideeffect, self.session, "http://eventws", db_bufsize=self.db_buf_size)

# assert we got the same result as above:

assert len(self.session.query(Event).all()) == len(pd.unique(data['id'])) == 2

assert len(data) == 2

# and since first response is 413, that having split the request into two, the

# second response is our URLError (we could test it better, anyway):

assert "blabla23___" in self.log_msg()

关注