def test_sort_data_check_status(self):
reputation = Reputation(self.df)
converted_df = reputation.df
status = converted_df.status.values
status = list(status)
expected = [0, 0, 0, 1, 1, 0, 0, 1, 0]
self.assertListEqual(status, expected, message_sort.format('status'))
def test_sort_data_check_id(self):
reputation = Reputation(self.df)
converted_df = reputation.df
ids = converted_df.id.values
ids = list(ids)
expected = ['555444******0777_2022-09']*5 + ['666555******0666_2022-10']*4
self.assertListEqual(ids, expected, message_sort.format('id'))
def test_sort_data_check_date_second(self):
reputation = Reputation(self.df)
converted_df = reputation.df
date_seconds = converted_df.date_seconds.values
date_seconds = list(date_seconds)
expected = [1577898731, 1578671420, 1579514465, 1583842210, 1586520855, 1546345212, 1589119861
,1589123523, 1589124725]
self.assertListEqual(date_seconds, expected, message_sort.format('date_second'))
def test_sort_data_check_date_only(self):
reputation = Reputation(self.df)
converted_df = reputation.df
date_only = converted_df.date_only.values
date_only = list(date_only)
expected = ['2020-01-01', '2020-01-10', '2020-01-20', '2020-03-10', '2020-04-10', '2019-01-01'
,'2020-05-10', '2020-05-10', '2020-05-10']
self.assertListEqual(date_only, expected, message_sort.format('date_only'))
def test_sort_data_check_date(self):
reputation = Reputation(self.df)
converted_df = reputation.df
dates = converted_df.date.values
dates = list(dates)
expected = ['2020-01-01 17:12:11', '2020-01-10 15:50:20', '2020-01-20 10:01:05', '2020-03-10 12:10:10'
,'2020-04-10 12:14:15', '2019-01-01 12:20:12', '2020-05-10 14:11:01', '2020-05-10 15:12:03', '2020-05-10 15:32:05']
self.assertListEqual(dates, expected, message_sort.format('date'))
def test_sort_data_check_date_cb_second(self):
# TOO test failed after rename field cb_second -> fa
reputation = Reputation(self.df)
converted_df = reputation.df
date_cb_seconds = converted_df.date_cb_seconds.values
date_cb_seconds = list(date_cb_seconds)
expected = [-9223372037,-9223372037, -9223372037, 1585699200, 1588291200, -9223372037, -9223372037
, 1590969600, -9223372037]
self.assertListEqual(date_cb_seconds, expected, message_sort.format('date_cb_second'))
def test_is_new_factor(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
is_new = list(df_factors.is_new.values)
expected = [1, 1, 0, 0, 0, 0, 0, 0, 0]
self.assertListEqual(is_new, expected, message_factor.format('is_first'))
def test_is_quick_factor(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
is_quick = list(df_factors.is_quick.values)
expected = [-1, -1, 0, 0, 0, 0, 0, 0, 1]
self.assertListEqual(is_quick, expected, message_factor.format('is-quick'))
def test_n_bad_factor(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
n_bad = list(df_factors.n_bad.values)
expected = [0, 0, 0, 0, 0, 1, 0, 0, 0]
self.assertListEqual(n_bad, expected, message_factor.format('n_bad'))
def test_n_grey_factor(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
n_grey = list(df_factors.n_grey.values)
expected = [0, 0, 0, 0, 3, 3, 1, 1, 1]
self.assertListEqual(n_grey, expected, message_factor.format('n_grey'))
def test_n_previous_factor(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
n_previous = list(df_factors.n_previous.values)
expected = [0, 0, 1, 2, 3, 4, 2, 1, 3]
self.assertListEqual(n_previous, expected, message_factor.format('n_previous'))
def test_create_reputation_factors(self):
reputation = Reputation(self.df)
dt = reputation.create_reputation_factors()
expected = DataForTests.data_reputation_factors
self.assertDictEqual(dt.to_dict(), expected, message_factor.format('create_reputation_factors'))
def test_amount_dev(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
amount_dev = list(df_factors.amount_dev.values)
expected = [-1, -1, 8, 17, 13, 13, 67, 20, 2]
self.assertListEqual(amount_dev, expected, message_factor.format('amount_dev'))
def test_delta_days(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
delta_days = list(df_factors.delta_days.values)
expected = [-1, -1, 214, 234, 1202, 744, 1, 11881, 0]
self.assertListEqual(delta_days, expected, message_factor.format('delta_days'))
def test_delta_sec(self):
reputation = Reputation(self.df)
df_factors = reputation.create_reputation_factors()
delta_sec = list(df_factors.delta_sec.values)
expected = [-1, -1, 772689, 843045, 4327745, 2678645, 3662, 42774649, 1202]
self.assertListEqual(delta_sec, expected, message_factor.format('delta_sec'))