def test_input_mixed_in_np_max():
obj = np.array([30, 53.0, "31", True, 32])
df = DataFrame(obj, colindex=["AGE"], rowindex=["A", "B", "C", "D", "E"])
expected_output = []
actual_output = df.max()
assert actual_output == expected_output
def test_input_int_in_np_max():
obj = np.array([30, 53, 31, 47, 32])
df = DataFrame(obj, colindex=["AGE"], rowindex=["A", "B", "C", "D", "E"])
expected_output = [53]
actual_output = df.max()
assert actual_output == expected_output
def test_input_int_in_list_of_lists_max():
obj = [[30, 53, 31, 47, 32], [4, 10, 2, 5, 4]]
df = DataFrame(
obj, colindex=["AGE", "ALBUMS"], rowindex=["A", "B", "C", "D", "E"]
)
expected_output = [53, 10]
actual_output = df.max()
assert actual_output == expected_output
def test_input_int_in_dict_of_np_max():
obj = {
"age": np.array([30, 53, 31, 47, 32]),
"albums": np.array([4, 10, 2, 5, 4]),
}
df = DataFrame(obj,
colindex=["AGE", "ALBUMS"],
rowindex=["A", "B", "C", "D", "E"])
expected_output = [53, 10]
actual_output = df.max()
assert actual_output == expected_output
def test_max_df():
data = {
"value1": [1, 2, 3, 4, 5, 6],
"value2": [2, 2, 2, 2, 2, 2],
"value3": [1.1, 2.2, 3.3, 4.4, 5.5, 6.6],
"value4": ["a", "b", "c", "d", "e", "f"],
}
expected_output = np.array([6, 2, 6.6, None])
df = DataFrame(data)
max_output = df.max()
print(max_output)
print(expected_output)
assert max_output.all() == expected_output.all()
def test_input_mixed_in_dict_of_np_max():
obj = {
"age": np.array([30.1, 53.1, 31.1, 47.1, 32.1]),
"albums": np.array([4, 10, 2, 5, 4]),
"C": np.array(["a", "b", "c", "d", "e"]),
"D": np.array([True, False, True, True, False]),
}
df = DataFrame(
obj,
colindex=["AGE", "ALBUMS", "C", "D"],
rowindex=["A", "B", "C", "D", "E"],
)
expected_output = [53.1, 10, True]
actual_output = df.max()
assert actual_output == expected_output
def test_input_mixed_in_list_of_lists_max():
obj = [
[30.1, 53.1, 31.1, 47.1, 32.1],
[4, 10, 2, 5, 4],
["a", "b", "c", "d", "e"],
[True, False, True, True, False],
]
df = DataFrame(
obj,
colindex=["AGE", "ALBUMS", "C", "D"],
rowindex=["A", "B", "C", "D", "E"],
)
expected_output = [53.1, 10, True]
actual_output = df.max()
assert actual_output == expected_output
def test_max():
dictionary = {
"pet": np.array(["cat", "dog", "mouse"]),
"age": np.array([1, 2, 3]),
"weight": np.array([1.0, 2.0, 3.0]),
"sick": np.array([True, True, False]),
}
df = DataFrame(dictionary)
max_collector = []
columns = []
for key in dictionary:
if dictionary[key].dtype == "float64" or \
dictionary[key].dtype == "int32":
columns.append(key)
max_collector.append(max(dictionary[key]))
expected_output = max_collector
output = df.max()
assert output == expected_output
def test_max(dictionary, expected):
myDF = DataFrame(dictionary)
assert myDF.max() == expected
