diff --git a/tests/froude_test.py b/tests/froude_test.py index 531cddc..08ead59 100644 --- a/tests/froude_test.py +++ b/tests/froude_test.py @@ -1,3 +1,4 @@ +import pandas as pd import pytest from scaling import froude @@ -16,6 +17,16 @@ def test_froude_time(): def test_froude_force(): assert froude.model_to_proto(1, 10, 'N', 'N') == 1000 +def test_froude_time_index_dataframe(): + df_model = pd.DataFrame(index=[2], data=[20]) + df_proto = froude.model_to_proto(df_model, 16, 'mm', 'm', 's', 's') + assert df_proto.index.values[0] == 8 + +def test_froude_mass_index_dataframe(): + df_proto = pd.DataFrame(index=[4], data=[16]) + df_model = froude.proto_to_model(df_proto, 2, 'kN', 'N', 'm', 'm') + assert df_model.index.values[0] == 2 + def test_unit_dimensions(): assert froude.dimensions('m') == 'L^1' assert froude.dimensions('kg') == 'M^1'