예제 #1
0
def test_check_num_columns_a():
    mock_dataset = [
        [1, "a"],
        [2, "b"],
        [3, "c"]
    ]
    assert check_num_columns(mock_dataset) == 2
예제 #2
0
def test_check_num_columns_b():
    mock_dataset = np.arange(9).reshape(-1, 3)
    assert check_num_columns(mock_dataset) == 3
예제 #3
0
def test_check_num_columns_a():
    mock_dataset = [[1, "a"], [2, "b"], [3, "c"]]
    assert check_num_columns(mock_dataset) == 2
예제 #4
0
def test_check_num_columns_b():
    mock_dataset = np.arange(9).reshape(-1, 3)
    assert check_num_columns(mock_dataset) == 3