def test_input_int_in_np_min(): obj = np.array([30, 53, 31, 47, 32]) df = DataFrame(obj, colindex=["AGE"], rowindex=["A", "B", "C", "D", "E"]) expected_output = [30] actual_output = df.min() assert actual_output == expected_output
def test_input_int_in_dict_of_lists_min(): obj = {"age": [30, 53, 31, 47, 32], "albums": [4, 10, 2, 5, 4]} df = DataFrame(obj, colindex=["AGE", "ALBUMS"], rowindex=["A", "B", "C", "D", "E"]) expected_output = [30, 2] actual_output = df.min() assert actual_output == expected_output
def test_min_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([1, 2, 1.1, None]) df = DataFrame(data) min_output = df.min() print(min_output) print(expected_output) assert min_output.all() == expected_output.all()
def test_input_mixed_in_dict_of_np_min(): 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 = [30.1, 2, False] actual_output = df.min() assert actual_output == expected_output
def test_input_mixed_in_list_of_lists_min(): 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 = [30.1, 2, False] actual_output = df.min() assert actual_output == expected_output
def test_min(): 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) min_collector = [] columns = [] for key in dictionary: if dictionary[key].dtype == "float64" or dictionary[key].dtype == "int32": columns.append(key) min_collector.append(min(dictionary[key])) expected_output = min_collector output = df.min() assert output == expected_output
def test_min(dictionary, expected): myDF = DataFrame(dictionary) assert myDF.min() == expected