Format code, Refactor test for readability

Signed-off-by: Seunghwan Hong <[email protected]>
apache · Jul 17, 2022 · 3863c33 · 3863c33
1 parent e2e4f37
commit 3863c33
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/sdks/python/apache_beam/io/gcp/bigquery_tools.py b/sdks/python/apache_beam/io/gcp/bigquery_tools.py
@@ -1537,7 +1537,10 @@ def encode(self, table_row):
     # to the programmer that they have used NAN/INF values.
     try:
       return json.dumps(
-          table_row, allow_nan=False, ensure_ascii=False, default=default_encoder).encode('utf-8')
+          table_row,
+          allow_nan=False,
+          ensure_ascii=False,
+          default=default_encoder).encode('utf-8')
     except ValueError as e:
       raise ValueError(
           '%s. %s. Row: %r' % (e, JSON_COMPLIANCE_ERROR, table_row))

diff --git a/sdks/python/apache_beam/io/gcp/bigquery_tools_test.py b/sdks/python/apache_beam/io/gcp/bigquery_tools_test.py
@@ -1068,10 +1068,10 @@ def test_invalid_json_neg_inf(self):
   def test_ensure_ascii(self):
     coder = RowAsDictJsonCoder()
     test_value = {'s': '🎉'}
-    should_be = b'{"s": "\xf0\x9f\x8e\x89"}'
+    output_value = b'{"s": "\xf0\x9f\x8e\x89"}'
+
+    self.assertEqual(output_value, coder.encode(test_value))
 
-    output_value = coder.encode(test_value)
-    self.assertEqual(output_value, should_be)
 
 @unittest.skipIf(HttpError is None, 'GCP dependencies are not installed')
 class TestJsonRowWriter(unittest.TestCase):