import random
import tqdm
from kstest.baselines import SpectralResidual, Luminol
from kstest.shift_detector import ShiftDetectorMSP
from kstest.utils import set_random_seed
from kstest.log import getLogger
import numpy as np
from kstest.dataset.ts_data import TSData, ReadNABDatasetWithTime
set_random_seed()
logger = getLogger(__name__)
class AbsExpTS:
SR = "SR"
SR_MOCHI = "SR_MOCHI"
SR_MOCHI_NL = "SR_MOCHI_NL"
SR_CORNER_SEARCH = "SR_CORNER_SEARCH"
SR_GRACE = "SR_GRACE"
SR_GREEDY = "SR_GREEDY"
SR_GRACE_ORIGIN = "SR_GRACE_ORIGIN"
SR_CORNER_SEARCH_ORIGIN = "SR_CORNER_SEARCH_ORIGIN"
L = "L"
L_MOCHI = "L_MOCHI"
L_MOCHI_NL = "L_MOCHI_NL"
L_CORNER_SEARCH = "L_CORNER_SEARCH"
L_CORNER_SEARCH_ORIGIN = "L_CORNER_SEARCH_ORIGIN"
L_GRACE = "L_GRACE"
import copy
import random
import time
from collections import defaultdict
import tqdm
import numpy as np
from kstest.log import getLogger
logging = getLogger(__name__)
RED = 1
BLUE = 2
VALUE_INDX = 0
TYPE_INDX = 1
INDICE_INDX = 2
POSITION_INDX = 1
RM_BALLS_INDX = 0
UPPER_BOUND_INDX = 0
LOWER_BOUND_INDX = 1
def compute_ks(x_1, x_2):
x = sorted([(i, RED, idx) for idx, i in enumerate(x_1)] + [(i, BLUE, idx) for idx, i in enumerate(x_2)],
key=lambda i: i[VALUE_INDX])
cdf_1 = {}
cdf_2 = {}
red_count = 0
blue_count = 0