def testFilterWithDictRHS(self):
(lhs, rhs, weights, rhs_pre, index, valid) = _filter_data(self.TS1, self.DICT1, None)
exp_lhs = self.TS1[1:]
exp_rhs1 = self.TS2[1:3]
exp_rhs2 = self.TS4[1:3]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs1, rhs["x1"])
self.tsAssertEqual(exp_rhs2, rhs["x2"])
def testFilterWithSeriesRHS(self):
(lhs, rhs, weights, rhs_pre,
index, valid) = _filter_data(self.TS1, {'x1': self.TS2}, None)
self.tsAssertEqual(self.TS1.astype(np.float64), lhs, check_names=False)
self.tsAssertEqual(self.TS2[:3].astype(np.float64), rhs['x1'],
check_names=False)
self.tsAssertEqual(self.TS2.astype(np.float64), rhs_pre['x1'],
check_names=False)
def testFilterWithDictRHS(self):
(lhs, rhs, rhs_pre, index, valid) = _filter_data(self.TS1, self.DICT1)
exp_lhs = self.TS1[1:]
exp_rhs1 = self.TS2[1:3]
exp_rhs2 = self.TS4[1:3]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs1, rhs['x1'])
self.tsAssertEqual(exp_rhs2, rhs['x2'])
def testFilterWithSeriesRHS3(self):
(lhs, rhs, weights, rhs_pre, index, valid) = _filter_data(self.TS3, {"x1": self.TS4}, None)
exp_lhs = self.TS3[2:3]
exp_rhs = self.TS4[2:3]
exp_rhs_pre = self.TS4[1:]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs, rhs["x1"])
self.tsAssertEqual(exp_rhs_pre, rhs_pre["x1"])
def testFilterWithDataFrameRHS(self):
(lhs, rhs, weights, rhs_pre,
index, valid) = _filter_data(self.TS1, self.DF1, None)
exp_lhs = self.TS1[1:]
exp_rhs1 = self.TS2[1:3]
exp_rhs2 = self.TS4[1:3]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs1, rhs['x1'])
self.tsAssertEqual(exp_rhs2, rhs['x2'])
def testFilterWithSeriesRHS3(self):
(lhs, rhs, weights, rhs_pre,
index, valid) = _filter_data(self.TS3, {'x1' : self.TS4}, None)
exp_lhs = self.TS3[2:3]
exp_rhs = self.TS4[2:3]
exp_rhs_pre = self.TS4[1:]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs, rhs['x1'])
self.tsAssertEqual(exp_rhs_pre, rhs_pre['x1'])
def testFilterWithDictRHS(self):
(lhs, rhs, weights, rhs_pre,
index, valid) = _filter_data(self.TS1, self.DICT1, None)
exp_lhs = self.TS1[1:].astype(np.float64)
exp_rhs1 = self.TS2[1:3].astype(np.float64)
exp_rhs2 = self.TS4[1:3].astype(np.float64)
self.tsAssertEqual(exp_lhs, lhs, check_names=False)
self.tsAssertEqual(exp_rhs1, rhs['x1'], check_names=False)
self.tsAssertEqual(exp_rhs2, rhs['x2'], check_names=False)
def testFilterWithSeriesRHS3(self):
(lhs, rhs, weights, rhs_pre,
index, valid) = _filter_data(self.TS3, {'x1': self.TS4}, None)
exp_lhs = self.TS3[2:3]
exp_rhs = self.TS4[2:3]
exp_rhs_pre = self.TS4[1:]
self.tsAssertEqual(exp_lhs, lhs, check_names=False)
self.tsAssertEqual(exp_rhs, rhs['x1'], check_names=False)
self.tsAssertEqual(exp_rhs_pre, rhs_pre['x1'], check_names=False)
def testFilterWithDictRHS(self):
(lhs, rhs, rhs_pre,
index, valid) = _filter_data(self.TS1, self.DICT1)
exp_lhs = self.TS1[1:]
exp_rhs1 = self.TS2[1:3]
exp_rhs2 = self.TS4[1:3]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs1, rhs['x1'])
self.tsAssertEqual(exp_rhs2, rhs['x2'])
def testFilterWithSeriesRHS3(self):
(lhs, rhs, rhs_pre,
index, valid) = _filter_data(self.TS3, {'x1' : self.TS4})
exp_lhs = self.TS3[2:3]
exp_rhs = self.TS4[2:3]
exp_rhs_pre = self.TS4[1:]
self.tsAssertEqual(exp_lhs, lhs)
self.tsAssertEqual(exp_rhs, rhs['x1'])
self.tsAssertEqual(exp_rhs_pre, rhs_pre['x1'])
def testFilterWithSeriesRHS2(self):
(lhs, rhs, weights, rhs_pre,
index, valid) = _filter_data(self.TS2, {'x1' : self.TS1}, None)
self.tsAssertEqual(self.TS2[:3], lhs)
self.tsAssertEqual(self.TS1, rhs['x1'])
self.tsAssertEqual(self.TS1, rhs_pre['x1'])
def testFilterWithSeriesRHS(self):
(lhs, rhs, weights, rhs_pre, index, valid) = _filter_data(self.TS1, {"x1": self.TS2}, None)
self.tsAssertEqual(self.TS1, lhs)
self.tsAssertEqual(self.TS2[:3], rhs["x1"])
self.tsAssertEqual(self.TS2, rhs_pre["x1"])
def testFilterWithSeriesRHS(self):
(lhs, rhs, rhs_pre, index,
valid) = _filter_data(self.TS1, {'x1': self.TS2})
self.tsAssertEqual(self.TS1, lhs)
self.tsAssertEqual(self.TS2[:3], rhs['x1'])
self.tsAssertEqual(self.TS2, rhs_pre['x1'])
def testFilterWithSeriesRHS(self):
(lhs, rhs, rhs_pre,
index, valid) = _filter_data(self.TS1, {'x1' : self.TS2})
self.tsAssertEqual(self.TS1, lhs)
self.tsAssertEqual(self.TS2[:3], rhs['x1'])
self.tsAssertEqual(self.TS2, rhs_pre['x1'])