def test_f6(self):
src = extract_f6(self.gb).sort_values(['Session ID', 'Item ID'])
dst = pd.DataFrame(np.array(
[[5, 214530776, 3], [10, 214820942, 1], [10, 214826810, 1],
[15, 214555903, 1], [15, 214547255, 2], [20, 214829282, 1],
[20, 214718203, 1], [20, 214819552, 1], [25, 214836761, 1],
[25, 214839313, 5], [30, 214820201, 2]],
dtype=np.int32),
columns=['Session ID', 'Item ID', 'Sequent Clicks'
]).sort_values(['Session ID', 'Item ID'])
np.testing.assert_array_equal(src, dst, 'Incorrect extraction of F6')
def test_f6(self):
src = extract_f6(self.gb).sort_values(['Session ID', 'Item ID'])
dst = pd.DataFrame(np.array([
[5, 214530776, 3],
[10, 214820942, 1],
[10, 214826810, 1],
[15, 214555903, 1],
[15, 214547255, 2],
[20, 214829282, 1],
[20, 214718203, 1],
[20, 214819552, 1],
[25, 214836761, 1],
[25, 214839313, 5],
[30, 214820201, 2]],
dtype=np.int32),
columns=['Session ID', 'Item ID', 'Sequent Clicks']
).sort_values(['Session ID', 'Item ID'])
np.testing.assert_array_equal(src, dst, 'Incorrect extraction of F6')
def test_p10(self):
src = extract_p10(extract_f6(self.gb))
dst = np.array([3, 1, 2, 1, 5, 2])
np.testing.assert_array_equal(src, dst, 'Incorrect extraction of P10')
def test_p10(self):
src = extract_p10(extract_f6(self.gb))
dst = np.array([3, 1, 2, 1, 5, 2])
np.testing.assert_array_equal(src, dst, 'Incorrect extraction of P10')
