Skip to content

Commit 15a69c1

Browse files
committed
lint
1 parent 63b3111 commit 15a69c1

File tree

1 file changed

+18
-7
lines changed

1 file changed

+18
-7
lines changed

metadata-ingestion/tests/unit/cli/dataset/test_dataset_cmd.py

+18-7
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ def invalid_value_yaml_file():
5252
if temp_file.exists():
5353
temp_file.unlink()
5454

55+
5556
@pytest.fixture
5657
def malformed_yaml_file():
5758
"""Creates a temporary malformed yaml file for testing."""
@@ -259,14 +260,17 @@ def test_dry_run_sync(self, mock_get_default_graph, test_yaml_file):
259260
assert not mock_get_default_graph.emit.called
260261

261262
@patch("datahub.cli.specific.dataset_cli.get_default_graph")
262-
def test_dry_run_sync_fail_bad_type(self, mock_get_default_graph, invalid_value_yaml_file):
263+
def test_dry_run_sync_fail_bad_type(
264+
self, mock_get_default_graph, invalid_value_yaml_file
265+
):
263266
mock_graph = MagicMock()
264267
mock_graph.exists.return_value = True
265268
mock_get_default_graph.return_value.__enter__.return_value = mock_graph
266269

267270
runner = CliRunner()
268271
result = runner.invoke(
269-
dataset, ["sync", "--dry-run", "--to-datahub", "-f", str(invalid_value_yaml_file)]
272+
dataset,
273+
["sync", "--dry-run", "--to-datahub", "-f", str(invalid_value_yaml_file)],
270274
)
271275

272276
# Verify
@@ -275,7 +279,9 @@ def test_dry_run_sync_fail_bad_type(self, mock_get_default_graph, invalid_value_
275279
assert "Type bad_type is not a valid primitive type" in result.output
276280

277281
@patch("datahub.cli.specific.dataset_cli.get_default_graph")
278-
def test_dry_run_sync_fail_missing_ref(self, mock_get_default_graph, test_yaml_file):
282+
def test_dry_run_sync_fail_missing_ref(
283+
self, mock_get_default_graph, test_yaml_file
284+
):
279285
mock_graph = MagicMock()
280286
mock_graph.exists.return_value = False
281287
mock_get_default_graph.return_value.__enter__.return_value = mock_graph
@@ -297,7 +303,9 @@ def test_run_sync(self, mock_get_default_graph, test_yaml_file):
297303
mock_get_default_graph.return_value.__enter__.return_value = mock_graph
298304

299305
runner = CliRunner()
300-
result = runner.invoke(dataset, ["sync", "--to-datahub", "-f", str(test_yaml_file)])
306+
result = runner.invoke(
307+
dataset, ["sync", "--to-datahub", "-f", str(test_yaml_file)]
308+
)
301309

302310
# Verify
303311
assert result.exit_code == 0
@@ -310,7 +318,9 @@ def test_run_sync_fail(self, mock_get_default_graph, invalid_value_yaml_file):
310318
mock_get_default_graph.return_value.__enter__.return_value = mock_graph
311319

312320
runner = CliRunner()
313-
result = runner.invoke(dataset, ["sync", "--to-datahub", "-f", str(invalid_value_yaml_file)])
321+
result = runner.invoke(
322+
dataset, ["sync", "--to-datahub", "-f", str(invalid_value_yaml_file)]
323+
)
314324

315325
# Verify
316326
assert result.exit_code != 0
@@ -338,9 +348,10 @@ def test_sync_from_datahub_fail(self, mock_get_default_graph, test_yaml_file):
338348
mock_get_default_graph.return_value.__enter__.return_value = mock_graph
339349

340350
runner = CliRunner()
341-
result = runner.invoke(dataset, ["sync", "--dry-run", "--from-datahub", "-f", str(test_yaml_file)])
351+
result = runner.invoke(
352+
dataset, ["sync", "--dry-run", "--from-datahub", "-f", str(test_yaml_file)]
353+
)
342354

343355
# Verify
344356
assert result.exit_code != 0
345357
assert "does not exist" in result.output
346-

0 commit comments

Comments
 (0)