def test_p11(self): src = extract_p11(extract_f7(self.gb)) dst = np.array([ 430260.0, unique_click_indicator, 246255.0, unique_click_indicator, 8674.0, 0.0 ]) np.testing.assert_array_equal(src, dst, 'Incorrect extraction of P11')
def test_f7(self): src = extract_f7(self.gb).sort_values(['Session ID', 'Item ID']) dst = pd.DataFrame( np.array([[5, 214530776, 430260], [10, 214820942, unique_click_indicator], [10, 214826810, unique_click_indicator], [15, 214555903, unique_click_indicator], [15, 214547255, 246255], [20, 214829282, unique_click_indicator], [20, 214718203, unique_click_indicator], [20, 214819552, unique_click_indicator], [25, 214836761, unique_click_indicator], [25, 214839313, 8674], [30, 214820201, 0]]), columns=['Session ID', 'Item ID', 'Time Difference']).sort_values(['Session ID', 'Item ID']) np.testing.assert_array_equal(src, dst, 'Incorrect extraction of F7')
def test_f7(self): src = extract_f7(self.gb).sort_values(['Session ID', 'Item ID']) dst = pd.DataFrame(np.array([ [5, 214530776, 430260], [10, 214820942, unique_click_indicator], [10, 214826810, unique_click_indicator], [15, 214555903, unique_click_indicator], [15, 214547255, 246255], [20, 214829282, unique_click_indicator], [20, 214718203, unique_click_indicator], [20, 214819552, unique_click_indicator], [25, 214836761, unique_click_indicator], [25, 214839313, 8674], [30, 214820201, 0]]), columns=['Session ID', 'Item ID', 'Time Difference'] ).sort_values(['Session ID', 'Item ID']) np.testing.assert_array_equal(src, dst, 'Incorrect extraction of F7')
def test_p11(self): src = extract_p11(extract_f7(self.gb)) dst = np.array([430260.0, unique_click_indicator, 246255.0, unique_click_indicator, 8674.0, 0.0]) np.testing.assert_array_equal(src, dst, 'Incorrect extraction of P11')