Example #1
0
 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"])
Example #2
0
 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)
Example #3
0
 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'])
Example #4
0
 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"])
Example #5
0
 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'])
Example #6
0
 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'])
Example #7
0
 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)
Example #8
0
 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)
Example #9
0
 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'])
Example #10
0
 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'])
Example #11
0
 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'])
Example #12
0
 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"])
Example #13
0
 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'])
Example #14
0
 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'])