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])
