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