def test_complex_series_fail(self):
import pandas as pd
for x in "a", 0, pd.DataFrame(), pd.DataFrame(0,
index=["r0", "r1"],
columns=["c0", "c1"]):
with self.assertRaises(TypeError):
util.check_type(x, type_=pd.Series)
def test_complex_series_succ_direct(self):
import pandas as pd
for x in "a", 0, pd.DataFrame(), pd.DataFrame(0,
index=["r0", "r1"],
columns=["c0", "c1"]):
self.assertIs(util.check_type(x, type_=pd.Series, check_not=True),
x)
def test_int_series_succ_direct(self):
import pandas as pd
for x in pd.Series(["a", "b"]), pd.Series([3.14,
2.71]), pd.Series([0, 0.]):
self.assertIs(
util.check_type(x, type_=int, check_dtype=True,
check_not=True), x)
def test_primitive_int_succ_direct(self):
for x in "a", 3.14, (0, 1), [0, 1]:
self.assertEqual(util.check_type(x, type_=int, check_not=True), x)
def test_primitive_int_fail_direct(self):
for x in -128, -2, -1, 0, 1, 2, 172:
with self.assertRaises(TypeError):
util.check_type(x, type_=int, check_not=True)
def test_multi_fail(self):
types = {str, float, tuple, list}
x = 0
with self.assertRaises(TypeError):
util.check_type(x, type_=types)
def test_multi_succ(self):
types = {str, float, tuple, list}
for x in "a", 3.14, (0, 1), [0, 1]:
self.assertEqual(util.check_type(x, type_=types), x)
def test_int_df_succ_direct(self):
import pandas as pd
x = pd.DataFrame({"c0": {"r0": 0, "r1": 1}, "c1": {"r0": 2, "r1": 3.}})
self.assertIs(
util.check_type(x, type_=int, check_dtype=True, check_not=True), x)
def test_primitive_int_fail(self):
for x in "a", 3.14, (0, 1), [0, 1]:
with self.assertRaises(TypeError):
util.check_type(x, type_=int)
def test_int_series_fail_direct(self):
import pandas as pd
x = pd.Series([0, 1])
with self.assertRaises(TypeError):
util.check_type(x, type_=int, check_dtype=True, check_not=True)
def test_int_list_succ_direct(self):
import pandas as pd
for x in ["a", "b"], [3.14, 2.71], [0, 0.]:
self.assertEqual(
util.check_type(x, type_=int, check_dtype=True,
check_not=True), x)
def test_int_list_fail_direct(self):
x = [0, 1]
with self.assertRaises(TypeError):
util.check_type(x, type_=int, check_dtype=True, check_not=True)
def test_int_tuple_succ_direct(self):
import pandas as pd
for x in ("a", "b"), (3.14, 2.71), (0, 0.):
self.assertEqual(
util.check_type(x, type_=int, check_dtype=True,
check_not=True), x)
def test_complex_series_succ(self):
import pandas as pd
for x in pd.Series(dtype=object), pd.Series([0, 1]):
# pd.Series.__equals__() tests elementwise equality, we want to test identity
self.assertIs(util.check_type(x, type_=pd.Series), x)
def test_complex_series_fail_direct(self):
import pandas as pd
for x in pd.Series(dtype=object), pd.Series([0, 1]):
with self.assertRaises(TypeError):
util.check_type(x, type_=pd.Series, check_not=True)
def test_primitive_int_succ(self):
for x in -128, -2, -1, 0, 1, 2, 172:
self.assertEqual(util.check_type(x, type_=int), x)
def test_int_df_fail_direct(self):
import pandas as pd
x = pd.DataFrame({"c0": {"r0": 0, "r1": 1}, "c1": {"r0": 2, "r1": 3}})
with self.assertRaises(TypeError):
util.check_type(x, type_=int, check_dtype=True, check_not=True)
def test_multi_fail_direct(self):
types = [str, float, tuple, list]
for x in "a", 3.14, (0, 1), [0, 1]:
with self.assertRaises(TypeError):
util.check_type(x, type_=types, check_not=True)
def test_multi_succ_direct(self):
types = {str, float, tuple, list}
x = 0
self.assertEqual(util.check_type(x, type_=types, check_not=True), x)
