def test_series_astype_a(self): s1 = Series(['a', 'b', 'c']) s2 = s1.astype(object) self.assertEqual(s2.to_pairs(), ((0, 'a'), (1, 'b'), (2, 'c'))) self.assertTrue(s2.dtype == object) # we cannot convert to float with self.assertRaises(ValueError): s1.astype(float)
def test_series_astype_b(self): s1 = Series([1, 3, 4, 0]) s2 = s1.astype(bool) self.assertEqual(s2.to_pairs(), ((0, True), (1, True), (2, True), (3, False))) self.assertTrue(s2.dtype == bool)
def test_series_pickle_a(self): s1 = Series(range(6), index=list('abcdef')) s2 = Series((2, 3, 0, -1, 8, 6), index=list('abcdef')) s3 = s2.astype(bool) for series in (s1, s2, s3): pbytes = pickle.dumps(series) series_new = pickle.loads(pbytes) for v in series: # iter labels # this compares series objects self.assertFalse(series_new.values.flags.writeable) self.assertEqual(series_new.loc[v], series.loc[v])