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')