updated from 2.0 to 2.0.1 for tests

tests/test_charge_point.py

@@ -16,14 +16,14 @@
 )
 from ocpp.v16.datatypes import MeterValue, SampledValue
 from ocpp.v16.enums import Action, RegistrationStatus
-from ocpp.v20 import ChargePoint as cp_20
+from ocpp.v201 import ChargePoint as cp_201
 from ocpp.v201.call import SetNetworkProfilePayload
 from ocpp.v201.datatypes import NetworkConnectionProfileType
 from ocpp.v201.enums import OCPPInterfaceType, OCPPTransportType, OCPPVersionType
 
 
 def test_getters_should_not_be_called_during_routemap_setup():
-    class ChargePoint(cp_20):
+    class ChargePoint(cp_201):
         @property
         def foo(self):
             raise RuntimeError("this will be raised")
@@ -36,12 +36,12 @@ def foo(self):
 
 
 def test_multiple_classes_with_same_name_for_handler():
-    class ChargerA(cp_20):
+    class ChargerA(cp_201):
         @on(Action.Heartbeat)
         def heartbeat(self, **kwargs):
             pass
 
-    class ChargerB(cp_20):
+    class ChargerB(cp_201):
         @on(Action.Heartbeat)
         def heartbeat(self, **kwargs):
             pass

tests/test_messages.py

@@ -150,7 +150,7 @@ def test_validate_get_composite_profile_payload():
     validate_payload(message, ocpp_version="1.6")
 
 
-@pytest.mark.parametrize("ocpp_version", ["1.6", "2.0"])
+@pytest.mark.parametrize("ocpp_version", ["1.6", "2.0.1"])
 def test_validate_payload_with_valid_payload(ocpp_version):
     """
     Test if validate_payload doesn't return any exceptions when it's