Merge branch 'main' of https://github.com/Hamzu24/mosaic

Hamzu24 · Nov 20, 2024 · bc10d61 · bc10d61
2 parents 30521cc + 31f7b9e
commit bc10d61
Showing 1 changed file with 23 additions and 7 deletions.
diff --git a/packages/schema_wrapper/test/test_generated_classes.py b/packages/schema_wrapper/test/test_generated_classes.py
@@ -128,7 +128,21 @@ def test_weather_plot():
         rRange = [2, 10],
         width = 800
     )
-    print(to_dict(plot_spec))
+    assert({'colorDomain': '$domain', 
+            'colorRange': '$colors', 
+            'plot': [
+                {'data': {'filterBy': '$click', 'from': 'weather'}, 
+                 'fill': 'weather', 
+                 'mark': 'dot', 
+                 'r': 'precipitation', 
+                 'x': {'dateMonthDay': 'date'}, 
+                 'y': 'temp_max'}
+                ], 
+            'rDomain': 'Fixed', 
+            'rRange': [2, 10], 
+            'width': 800, 
+            'xTickFormat': '%b', 
+            'xyDomain': 'Fixed'} == to_dict(plot_spec))
 
 def test_stock_plot():
     plot_spec = Plot(
@@ -141,12 +155,14 @@ def test_stock_plot():
         width = 680,
         height = 200
     )
-    assert to_dict(plot_spec) == {
-        'height': 200, 
-        'plot': [{'data': {'from': 'aapl'}, 'mark': 'lineY', 'x': 'Date', 'y': 'Close'}], 
-        'width': 680
-        }
-
+    assert({'height': 200, 
+            'plot': [
+                {'data': {'from': 'aapl'}, 
+                 'mark': 'lineY', 
+                 'x': 'Date', 
+                 'y': 'Close'}
+                ], 
+            'width': 680} == to_dict(plot_spec))
 
 if __name__ == '__main__':
     pytest.main([__file__])